joj3_config_generator/lib/task.py

@@ -1,5 +1,5 @@
 import shlex
-from typing import Any, Dict, Tuple
+from typing import Tuple
 
 import rtoml
 
@@ -13,7 +13,7 @@ from joj3_config_generator.models.result import Stage as ResultStage
 from joj3_config_generator.models.task import Stage as TaskStage
 
 
-def remove_nulls(d: Dict[str, Any]) -> Dict[str, Any]:
+def remove_nulls(d: result.Config) -> result.Config:
     if isinstance(d, dict):
         return {k: remove_nulls(v) for k, v in d.items() if v is not None}
     elif isinstance(d, list):

tests/convert/test_convert_cases.py

@@ -4,4 +4,5 @@ from tests.convert.utils import load_case
 def test_basic() -> None:
     load_case("basic")
 
-test_basic()
+
+test_basic()

tests/convert/utils.py

@@ -29,5 +29,5 @@ def read_convert_files(
 def load_case(case_name: str) -> None:
     repo, task, expected_result = read_convert_files(case_name)
     result = convert(repo, task).model_dump(mode="json", by_alias=True)
-    
+
     assert result == expected_result