SS-1288 Pin mlflow version (#293)

Signed-off-by: Nikita Churikov <[email protected]>

Author: churnikov

apps/tasks.py

@@ -1,5 +1,5 @@
+import re
 import subprocess
-import uuid
 
 import yaml
 from celery import shared_task
@@ -17,6 +17,8 @@
 
 logger = get_logger(__name__)
 
+CHART_REGEX = re.compile(r"^(?P<chart>.+):(?P<version>.+)$")
+
 
 @app.task
 def delete_old_objects():
@@ -194,9 +196,15 @@ def deploy_resource(serialized_instance):
     logger.info("Deploying resource for instance %s", instance)
     values = instance.k8s_values
     release = values["subdomain"]
+    chart: str = instance.chart
     if "ghcr" in instance.chart:
         version = instance.chart.split(":")[-1]
         chart = "oci://" + instance.chart.split(":")[0]
+    elif chart.startswith("oci://"):
+        match = CHART_REGEX.match(chart)
+        if match:
+            chart = match.group("chart")
+            version = match.group("version")
     else:
         version = None
         chart = instance.chart

fixtures/apps_fixtures.json

@@ -39,7 +39,7 @@
   {
     "fields": {
       "category": "develop",
-      "chart": "oci://registry-1.docker.io/bitnamicharts/mlflow",
+      "chart": "oci://registry-1.docker.io/bitnamicharts/mlflow:2.5.2",
       "created_on": "2025-02-12T21:34:37.815Z",
       "description": "",
       "name": "MLFlow",