docs: rename enable_checkpoints to checkpoint_strategy (#257)

argilla-io · Jan 16, 2024 · 768fa56 · 768fa56
1 parent 28b60c1
commit 768fa56
Show file tree

Hide file tree

Showing 9 changed files with 9 additions and 9 deletions.
diff --git a/docs/snippets/technical-reference/pipeline/pipe_2.py b/docs/snippets/technical-reference/pipeline/pipe_2.py
@@ -10,6 +10,6 @@
     instruction_dataset,
     num_generations=2,
     batch_size=1,
-    enable_checkpoints=True,
+    checkpoint_strategy=True,
     display_progress_bar=True,
 )
diff --git a/docs/tutorials/pipeline-notus-instructions-preferences-legal.ipynb b/docs/tutorials/pipeline-notus-instructions-preferences-legal.ipynb
@@ -714,7 +714,7 @@
     "    instructions_dataset,  # type: ignore\n",
     "    num_generations=2,\n",
     "    batch_size=8,\n",
-    "    enable_checkpoints=True,\n",
+    "    checkpoint_strategy=True,\n",
     "    display_progress_bar=True,\n",
     ")"
    ]

diff --git a/examples/pipeline-accelerate-and-openai.py b/examples/pipeline-accelerate-and-openai.py
@@ -76,7 +76,7 @@ def get_current_device() -> int:
             inputs,  # type: ignore
             num_generations=2,
             batch_size=1,
-            enable_checkpoints=True,
+            checkpoint_strategy=True,
             display_progress_bar=True,
         )
         dataset = gather_object(dataset)

diff --git a/examples/pipeline-fn-ultrafeedback-labeller.py b/examples/pipeline-fn-ultrafeedback-labeller.py
@@ -41,7 +41,7 @@
         num_generations=2,
         batch_size=1,
         display_progress_bar=True,
-        enable_checkpoints=True,
+        checkpoint_strategy=True,
     )
     end = time.time()
     print("Elapsed", end - start)

diff --git a/examples/pipeline-fn-ultrafeedback.py b/examples/pipeline-fn-ultrafeedback.py
@@ -50,7 +50,7 @@
         dataset,  # type: ignore
         num_generations=2,
         batch_size=1,
-        enable_checkpoints=True,
+        checkpoint_strategy=True,
         display_progress_bar=True,
     )
     end = time.time()

diff --git a/examples/pipeline-llamacpp-and-openai-process.py b/examples/pipeline-llamacpp-and-openai-process.py
@@ -71,7 +71,7 @@ def load_openai_llm(task: "Task") -> "LLM":
         dataset,  # type: ignore
         num_generations=2,
         batch_size=1,
-        enable_checkpoints=True,
+        checkpoint_strategy=True,
         display_progress_bar=False,
     )
 

diff --git a/examples/pipeline-llamacpp-and-openai.py b/examples/pipeline-llamacpp-and-openai.py
@@ -51,7 +51,7 @@
         dataset,  # type: ignore
         num_generations=2,
         batch_size=1,
-        enable_checkpoints=True,
+        checkpoint_strategy=True,
         display_progress_bar=True,
     )
 

diff --git a/examples/pipeline-transformers-and-openai.py b/examples/pipeline-transformers-and-openai.py
@@ -58,7 +58,7 @@
         dataset,  # type: ignore
         num_generations=2,
         batch_size=1,
-        enable_checkpoints=True,
+        checkpoint_strategy=True,
         display_progress_bar=True,
     )
 

diff --git a/examples/pipeline-vllm-and-openai.py b/examples/pipeline-vllm-and-openai.py
@@ -49,7 +49,7 @@
         dataset,  # type: ignore
         num_generations=2,
         batch_size=1,
-        enable_checkpoints=True,
+        checkpoint_strategy=True,
         display_progress_bar=True,
     )