Change name from LogIXArgument to LogIXArguments

logix-project · May 31, 2024 · 0f465d6 · 0f465d6
1 parent 1871908
commit 0f465d6
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/examples/huggingface/bert_influence.py b/examples/huggingface/bert_influence.py
@@ -7,7 +7,7 @@
 from transformers import default_data_collator, Trainer, TrainingArguments
 
 import logix
-from logix.huggingface import patch_trainer, LogIXArgument
+from logix.huggingface import patch_trainer, LogIXArguments
 from bert_utils import construct_model, get_datasets, set_seed
 
 
@@ -28,7 +28,7 @@ def main():
     model.eval()
     train_dataset = get_datasets(args.data_name)[-1]
 
-    logix_args = LogIXArgument(
+    logix_args = LogIXArguments(
         project=args.project,
         config=args.config_path,
         lora=True,

diff --git a/examples/huggingface/bert_log.py b/examples/huggingface/bert_log.py
@@ -6,7 +6,7 @@
 from transformers import default_data_collator, Trainer, TrainingArguments
 
 import logix
-from logix.huggingface import patch_trainer, LogIXArgument
+from logix.huggingface import patch_trainer, LogIXArguments
 from bert_utils import construct_model, get_datasets, set_seed
 
 
@@ -27,7 +27,7 @@ def main():
     model.eval()
     train_dataset = get_datasets(args.data_name)[1]
 
-    logix_args = LogIXArgument(
+    logix_args = LogIXArguments(
         project=args.project,
         config=args.config_path,
         lora=True,

diff --git a/examples/huggingface/gpt_influence.py b/examples/huggingface/gpt_influence.py
@@ -7,7 +7,7 @@
 from transformers import default_data_collator, Trainer, TrainingArguments
 
 import logix
-from logix.huggingface import patch_trainer, LogIXArgument
+from logix.huggingface import patch_trainer, LogIXArguments
 from gpt_utils import construct_model, get_datasets, set_seed
 
 
@@ -26,7 +26,7 @@ def main():
     model.eval()
     train_dataset = get_datasets()[-1]
 
-    logix_args = LogIXArgument(
+    logix_args = LogIXArguments(
         project=args.project,
         config=args.config_path,
         lora=True,

diff --git a/examples/huggingface/gpt_log.py b/examples/huggingface/gpt_log.py
@@ -6,7 +6,7 @@
 from transformers import default_data_collator, Trainer, TrainingArguments
 
 import logix
-from logix.huggingface import patch_trainer, LogIXArgument
+from logix.huggingface import patch_trainer, LogIXArguments
 from gpt_utils import construct_model, get_datasets, set_seed
 
 
@@ -25,7 +25,7 @@ def main():
     model.eval()
     train_dataset = get_datasets()[0]
 
-    logix_args = LogIXArgument(
+    logix_args = LogIXArguments(
         project=args.project,
         config=args.config_path,
         lora=True,