🎨 Move artifact path to use os path join

Signed-off-by: gkumbhat <[email protected]> Signed-off-by: gkumbhat <[email protected]>
gkumbhat · Aug 7, 2023 · 6907c9d · 6907c9d
1 parent 4fed554
commit 6907c9d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/run_fine_tuning.py b/examples/run_fine_tuning.py
@@ -342,9 +342,9 @@ def export_model_preds(preds_file, predictions, validation_stream):
         # HACK: export args.output_dir as MODEL_NAME for TGIS
         # container to pick up automatically
         os.environ["MODEL_DIR"] = os.path.dirname(args.output_dir)
-        os.environ[
-            "MODEL_NAME"
-        ] = f"/models/{os.path.basename(args.output_dir)}/artifacts"
+        os.environ["MODEL_NAME"] = os.path.join(
+            "models", os.path.basename(args.output_dir), "artifacts"
+        )
 
         loaded_model = load_model(is_distributed=True, model_path=args.output_dir)