Merge pull request #10 from paplorinc/paplorinc-patch-1

Fix colab incompatibilty
zetavg · Apr 19, 2023 · e302cf6 · e302cf6
2 parents acb1b11 + 81aa30e
commit e302cf6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/llama_lora/lib/csv_logger.py b/llama_lora/lib/csv_logger.py
@@ -5,7 +5,7 @@
 import re
 import secrets
 from pathlib import Path
-from typing import Any, List
+from typing import Any, List, Union
 
 class CSVLogger(FlaggingCallback):
     """
@@ -25,6 +25,8 @@ def __init__(self):
 
     def setup(
         self,
+        components: List[Any],
+        flagging_dir: Union[str, Path],
         components,
         flagging_dir,
     ):
@@ -36,7 +38,7 @@ def flag(
         self,
         flag_data: List[Any],
         flag_option: str = "",
-        username=None,
+        username: Union[str, None] = None,
         filename="log.csv",
     ) -> int:
         flagging_dir = self.flagging_dir