tests/commands/execution/test_run.py

@@ -2,7 +2,7 @@
 import yaml
 
 from tests.commands.run_test_utils import ALTERNATIVE_YAML, RunAPIMock, RunTestSetup
-from tests.fixture_data import CONFIG_YAML, PROJECT_DATA, KUBE_RESOURCE_YAML
+from tests.fixture_data import CONFIG_YAML, KUBE_RESOURCE_YAML, PROJECT_DATA
 from valohai_cli.commands.execution.run import run
 from valohai_cli.ctx import get_project
 from valohai_cli.models.project import Project
@@ -390,4 +390,4 @@ def test_kube_step_override_device_empty(run_test_setup_kube):
                 }
             }
         }
-    }
+    }

valohai_cli/commands/execution/run/frontend_command.py

@@ -133,7 +133,6 @@ def run(
     if autorestart:
         runtime_config["autorestart"] = autorestart
 
-
     if k8s_devices and k8s_device_none:
         raise click.UsageError('--k8s-device=(...) and --k8s-device-none cannot be used together. '
                                'Using --k8s-device=(...) will discard all Kubernetes device default values on its own.')

valohai_cli/utils/__init__.py

@@ -5,7 +5,7 @@
 import string
 import unicodedata
 import webbrowser
-from typing import Any, Dict, Iterable, Iterator, Tuple, Union, TypeVar, Callable
+from typing import Any, Callable, Dict, Iterable, Iterator, Tuple, Union
 
 import click
 
@@ -132,13 +132,13 @@ def sanitize_option_name(name: str) -> str:
     return re.sub(r'[^-a-z0-9]+', '-', name, flags=re.IGNORECASE).strip('-')
 
 
-ValueType = TypeVar('ValueType', bound=str)
-
+# Once mypy supports using TypeVars in default arguments, we can make a more specific type for coerce.
+# Mypy issue: https://github.com/python/mypy/issues/3737
 def parse_environment_variable_strings(
     envvar_strings: Iterable[str],
     *,
-    coerce: Callable[[str], ValueType] = str
-) -> Dict[str, ValueType]:
+    coerce: Callable[[str], Any] = str
+) -> Dict[str, Any]:
     """
     Parse a list of environment variable strings into a dict.
     """