Fix type hints (#35)

bagherilab · Feb 16, 2023 · 15249f7 · 15249f7
1 parent f9936a1
commit 15249f7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/container_collection/batch/register_batch_job.py b/src/container_collection/batch/register_batch_job.py
@@ -4,7 +4,7 @@
 
 
 @task
-def register_batch_job(job_definition: dict) -> None:
+def register_batch_job(job_definition: dict) -> str:
     client = boto3.client("batch")
 
     response = client.describe_job_definitions(

diff --git a/src/container_collection/manifest/summarize_manifest_files.py b/src/container_collection/manifest/summarize_manifest_files.py
@@ -7,7 +7,7 @@
 
 @task
 def summarize_manifest_files(
-    manifest: pd.DataFrame, name: str, conditions: list[dict], seeds: list[str]
+    manifest: pd.DataFrame, name: str, conditions: list[dict], seeds: list[int]
 ) -> str:
     condition_keys = [f"{name}_{condition['key']}" for condition in conditions]
     manifest_keys = manifest.set_index("KEY").filter(regex=f"^{name}", axis="index").reset_index()