Set ER name

VertaAI · Sep 19, 2023 · 5a40e49 · 5a40e49 · github-actions · Sep 19, 2023
1 parent d83ef0d
commit 5a40e49
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/client/verta/verta/finetune/__init__.py b/client/verta/verta/finetune/__init__.py
@@ -8,7 +8,7 @@
 from ._lora_config import LoraConfig
 
 
-_PROJECT_NAME_SUFFIX = " Fine-Tuning"  # append to registered model name
+_TRACKING_NAME_SUFFIX = " Fine-Tuning"  # append to RM and RMV names for project and run
 _EXPERIMENT_NAME = "Fine-Tuning"
 _TRAIN_DATASET_NAME = "train"
 _EVAL_DATASET_NAME = "eval"

diff --git a/client/verta/verta/registry/entities/_modelversion.py b/client/verta/verta/registry/entities/_modelversion.py
@@ -1839,7 +1839,7 @@ def finetune(
             self._conf,
             ctx,
             name=destination_registered_model.name
-            + verta.finetune._PROJECT_NAME_SUFFIX,
+            + verta.finetune._TRACKING_NAME_SUFFIX,
         )
         ctx.expt = Client._get_or_create_experiment(
             self._conn,
@@ -1851,6 +1851,8 @@ def finetune(
             self._conn,
             self._conf,
             ctx,
+            # TODO: make this match the RMV when `name` is None
+            name=name + verta.finetune._TRACKING_NAME_SUFFIX if name else None,
             attrs={verta.finetune._FINETUNE_ATTR_KEY: True},
         )