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

llama_lora/lib/csv_logger.py

@@ -5,7 +5,7 @@ import os
 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 @@ class CSVLogger(FlaggingCallback):
 
     def setup(
         self,
+        components: List[Any],
+        flagging_dir: Union[str, Path],
         components,
         flagging_dir,
     ):
@@ -36,7 +38,7 @@ class CSVLogger(FlaggingCallback):
         self,
         flag_data: List[Any],
         flag_option: str = "",
-        username=None,
+        username: Union[str, None] = None,
         filename="log.csv",
     ) -> int:
         flagging_dir = self.flagging_dir