Fixing py docs: arguments for client APIs (mlflow#2595)

Force merging since failing tests were unrelated. docs test passed.
MishaVeldhoen · Mar 17, 2020 · 21d3f40 · 21d3f40
1 parent 79f6a05
commit 21d3f40
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/mlflow/tracking/_model_registry/client.py b/mlflow/tracking/_model_registry/client.py
@@ -130,7 +130,7 @@ def transition_model_version_stage(self, name, version, stage, archive_existing_
 
         :param name: Registered model name.
         :param version: Registered model version.
-        :param new_stage: New desired stage for this model version.
+        :param stage: New desired stage for this model version.
         :param archive_existing_versions: If this flag is set, all existing model
                versions in the stage will be atomically moved to the "archived" stage.
 

diff --git a/mlflow/tracking/client.py b/mlflow/tracking/client.py
@@ -498,9 +498,7 @@ def transition_model_version_stage(self, name, version, stage):
 
         :param name: Registered model name.
         :param version: Registered model version.
-        :param new_stage: New desired stage for this model version.
-        :param archive_existing_versions: If this flag is set, all existing model
-               versions in the stage will be atomically moved to the "archived" stage.
+        :param stage: New desired stage for this model version.
 
         :return: A single :py:class:`mlflow.entities.model_registry.ModelVersion` object.
         """