add float to type anno

Matgenix · Jul 25, 2024 · 9e403bd · 9e403bd
1 parent 120bfaf
commit 9e403bd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/qtoolkit/io/pbs.py b/src/qtoolkit/io/pbs.py
@@ -353,7 +353,7 @@ def _convert_memory_str(memory: str | None) -> int | None:
     }
 
     @staticmethod
-    def _convert_time_to_str(time: int | timedelta) -> str:
+    def _convert_time_to_str(time: int | float | timedelta) -> str:
         if not isinstance(time, timedelta):
             time = timedelta(seconds=time)
 

diff --git a/src/qtoolkit/io/slurm.py b/src/qtoolkit/io/slurm.py
@@ -539,7 +539,7 @@ def _convert_memory_str(memory: str | None) -> int | None:
         return v * (1024 ** power_labels[units])
 
     @staticmethod
-    def _convert_time_to_str(time: int | timedelta) -> str:
+    def _convert_time_to_str(time: int | float | timedelta) -> str:
         if not isinstance(time, timedelta):
             time = timedelta(seconds=time)