From 60668f86df283d3c343812ef8ba473557eef3de8 Mon Sep 17 00:00:00 2001
From: Nuvole <jon-lee@sjtu.edu.cn>
Date: Sat, 9 Nov 2024 14:54:39 +0800
Subject: [PATCH] fix: type annotation

---
 joj3_config_generator/lib/task.py   | 4 ++--
 tests/convert/test_convert_cases.py | 3 ++-
 tests/convert/utils.py              | 2 +-
 3 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/joj3_config_generator/lib/task.py b/joj3_config_generator/lib/task.py
index f562b9b..fdfab1f 100644
--- a/joj3_config_generator/lib/task.py
+++ b/joj3_config_generator/lib/task.py
@@ -1,12 +1,12 @@
 import shlex
-from typing import Any, Dict, Tuple
+from typing import Tuple
 
 import rtoml
 
 from joj3_config_generator.models import joj1, repo, result, task
 
 
-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):
diff --git a/tests/convert/test_convert_cases.py b/tests/convert/test_convert_cases.py
index c3ac6cb..6aa8a0c 100644
--- a/tests/convert/test_convert_cases.py
+++ b/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()
\ No newline at end of file
+
+test_basic()
diff --git a/tests/convert/utils.py b/tests/convert/utils.py
index 0d7e395..8a2c3f1 100644
--- a/tests/convert/utils.py
+++ b/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