mlrun 1.5.0rc16__py3-none-any.whl → 1.5.0rc17__py3-none-any.whl

mlrun/api/api/endpoints/feature_store.py

@@ -409,6 +409,11 @@ async def ingest_feature_set(
             auth_info,
         )
 
+    # Set the run db instance with the current db session
+    await run_in_threadpool(
+        feature_set._override_run_db,
+        mlrun.api.api.utils.get_run_db_instance(db_session),
+    )
     if ingest_parameters.targets:
         data_targets = [
             DataTargetBase.from_dict(data_target.dict())

mlrun/api/crud/model_monitoring/model_endpoints.py

@@ -227,6 +227,11 @@ class ModelEndpoints:
             timestamp_key=mlrun.common.schemas.model_monitoring.EventFieldType.TIMESTAMP,
             description=f"Monitoring feature set for endpoint: {model_endpoint.spec.model}",
         )
+
+        # Set the run db instance with the current db session
+        feature_set._override_run_db(
+            mlrun.api.api.utils.get_run_db_instance(db_session)
+        )
         feature_set.metadata.project = model_endpoint.metadata.project
 
         feature_set.metadata.labels = {

mlrun/feature_store/feature_set.py

@@ -424,6 +424,9 @@ class FeatureSet(ModelObj):
         else:
             return mlrun.get_run_db()
 
+    def _override_run_db(self, run_db):
+        self._run_db = run_db
+
     def get_target_path(self, name=None):
         """get the url/path for an offline or specified data target"""
         target = get_offline_target(self, name=name)

mlrun/feature_store/retrieval/job.py

@@ -167,7 +167,7 @@ class RemoteVectorResponse:
             if self.with_indexes:
                 columns += self.vector.status.index_keys
                 if self.vector.status.timestamp_key is not None:
-                    columns.append(self.vector.status.timestamp_key)
+                    columns.insert(0, self.vector.status.timestamp_key)
 
         file_format = kwargs.get("format")
         if not file_format:

mlrun/model_monitoring/batch.py

@@ -356,7 +356,14 @@ class VirtualDrift:
             # Calculate the feature's drift mean:
             tvd = results[TotalVarianceDistance.NAME]
             hellinger = results[HellingerDistance.NAME]
-            if not tvd or not hellinger:
+            if tvd is None or hellinger is None:
+                logger.warning(
+                    "Can't calculate drift for this feature because at least one of the required"
+                    "statistical metrics is missing",
+                    feature=feature,
+                    tvd=tvd,
+                    hellinger=hellinger,
+                )
                 continue
             metrics_results_dictionary = (tvd + hellinger) / 2
             # Decision rule for drift detection:

mlrun/projects/pipelines.py

@@ -52,7 +52,7 @@ def get_workflow_engine(engine_kind, local=False):
         elif engine_kind == "remote":
             raise mlrun.errors.MLRunInvalidArgumentError(
                 "cannot run a remote pipeline locally using `kind='remote'` and `local=True`. "
-                "in order to run a local pipeline remotely, please use `engine='remote: local'` instead"
+                "in order to run a local pipeline remotely, please use `engine='remote:local'` instead"
             )
         return _LocalRunner
     if not engine_kind or engine_kind == "kfp":

mlrun/projects/project.py

@@ -1265,7 +1265,7 @@ class MlrunProject(ModelObj):
         :param name:          name of the workflow
         :param workflow_path: url/path for the workflow file
         :param embed:         add the workflow code into the project.yaml
-        :param engine:        workflow processing engine ("kfp" or "local")
+        :param engine:        workflow processing engine ("kfp", "local", "remote" or "remote:local")
         :param args_schema:   list of arg schema definitions (:py:class`~mlrun.model.EntrypointParam`)
         :param handler:       workflow function handler
         :param schedule:      ScheduleCronTrigger class instance or a standard crontab expression string
@@ -1285,8 +1285,9 @@ class MlrunProject(ModelObj):
                 f"Invalid 'workflow_path': '{workflow_path}'. Please provide a valid URL/path to a file."
             )
 
-        if image and engine not in ["remote"]:
-            logger.warning("Image is only relevant for remote workflows, ignoring it")
+        # engine could be "remote" or "remote:local"
+        if image and ((engine and "remote" in engine) or schedule):
+            logger.warning("Image is only relevant for 'remote' engine, ignoring it")
 
         if embed:
             if (
@@ -2496,6 +2497,8 @@ class MlrunProject(ModelObj):
         inner_engine = None
         if engine and engine.startswith("remote"):
             if ":" in engine:
+
+                # inner could be either kfp or local
                 engine, inner_engine = engine.split(":")
         elif workflow_spec.schedule:
             inner_engine = engine

mlrun/utils/version/version.json

@@ -1,4 +1,4 @@
 {
-  "git_commit": "7651a4c2cd1fb11dd6f068fdec3bf3cc6390ddea",
-  "version": "1.5.0-rc16"
+  "git_commit": "f429a0c8d769659dd17515b9056234b9d9561d86",
+  "version": "1.5.0-rc17"
 }

{mlrun-1.5.0rc16.dist-info → mlrun-1.5.0rc17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mlrun
-Version: 1.5.0rc16
+Version: 1.5.0rc17
 Summary: Tracking and config of machine learning runs
 Home-page: https://github.com/mlrun/mlrun
 Author: Yaron Haviv

{mlrun-1.5.0rc16.dist-info → mlrun-1.5.0rc17.dist-info}/RECORD

@@ -29,7 +29,7 @@ mlrun/api/api/endpoints/background_tasks.py,sha256=5SV4iDF-dZR878kp77VpANRb86BJY
 mlrun/api/api/endpoints/client_spec.py,sha256=lN2By-KGzHuzNTGiJk0RK3SM6ZUr3VLgsudsymrtZE8,1214
 mlrun/api/api/endpoints/clusterization_spec.py,sha256=0eSG7TSt647V7-A-uI5CLRItkVOBqIBM9ep_fiMFqPI,1175
 mlrun/api/api/endpoints/datastore_profile.py,sha256=yzW4VI7yYwWRAHieMcBch4IVz3WE2ctLrE2fN55ci2w,7475
-mlrun/api/api/endpoints/feature_store.py,sha256=mq6XVv9zaW7zU-XRHh8X2gTwk-z6a1QE3bcuOdR6jQU,28853
+mlrun/api/api/endpoints/feature_store.py,sha256=OwNnHw_yUZ6P4NZEwD2w4j2hks6xmBaGbTqi1Q8j75c,29045
 mlrun/api/api/endpoints/files.py,sha256=dFwmP40O2MXI1CmU3gJ1XrOVjSubpb0KJtguWJC5Bn4,6712
 mlrun/api/api/endpoints/frontend_spec.py,sha256=r8r6BSDpYcUpWKJX7zqXJPtYBZHzfZILd1w-SIikea8,5530
 mlrun/api/api/endpoints/functions.py,sha256=WSQWv0QXMoGgnEunSMQBbPFVig0H8hdbxkaGokUgrBQ,41111
@@ -73,7 +73,7 @@ mlrun/api/crud/model_monitoring/__init__.py,sha256=_rZmfvis_Qkhephv_SEgzqUVuC4bY
 mlrun/api/crud/model_monitoring/deployment.py,sha256=hHvknNfPJKBhC_QmxSEjojfvFUTkokw8ZTi8FBL8s1o,32279
 mlrun/api/crud/model_monitoring/grafana.py,sha256=7D-sUY3PjFh7_HjYe-KWryvepxO9zLArVC3U7IChuqA,17494
 mlrun/api/crud/model_monitoring/helpers.py,sha256=5K6y7hIjBdxY__rNfluINFgis1IJZQWuK-TjP26ypag,6114
-mlrun/api/crud/model_monitoring/model_endpoints.py,sha256=s1KwAAZ0sz6SY5XZT4eyyVKBTysnqHzEnLdtwd11aTU,31423
+mlrun/api/crud/model_monitoring/model_endpoints.py,sha256=WSVopB2hV2YFiyKTaXXBdpvBDt-zQ0Nv8qPrOiQ2T2c,31598
 mlrun/api/crud/runtimes/__init__.py,sha256=xY3wHC4TEJgez7qtnn1pQvHosi8-5UJOCtyGBS7FcGE,571
 mlrun/api/crud/runtimes/nuclio/__init__.py,sha256=xY3wHC4TEJgez7qtnn1pQvHosi8-5UJOCtyGBS7FcGE,571
 mlrun/api/crud/runtimes/nuclio/function.py,sha256=Tgca7uACPlyeoGiLkIHsX0d9SePy_a6pEW03ryGMdPc,18762
@@ -264,14 +264,14 @@ mlrun/db/nopdb.py,sha256=Z7Di4Tz7OXMZNq8clVZInKlY0QzHu5xoo4KN5soGw_E,13990
 mlrun/feature_store/__init__.py,sha256=n1F5m1svFW2chbE2dJdWzZJJiYS4E-y8PQsG9Q-F0lU,1584
 mlrun/feature_store/api.py,sha256=0-YFccl_p3c7jPizxJ2P1lcdtSkbhJhIIUHP8TRQay8,45276
 mlrun/feature_store/common.py,sha256=sl2pmkuv5w4KtTMjt0ky34Qps0iuoXNWbyYGuBkCxjc,12809
-mlrun/feature_store/feature_set.py,sha256=Kqtpasmdn6aypXZBXTrYGQDRLh7NUDya-JZOjkaNL6U,47422
+mlrun/feature_store/feature_set.py,sha256=oRZkMT4oHSb0Z3ViaHprPNXEBk8CrEYwOLVNO8FVQLo,47493
 mlrun/feature_store/feature_vector.py,sha256=3v0MFS4LvcFQ7cJA6-vPoa1tJ-qMIejnDuJhHGY7J5k,35012
 mlrun/feature_store/ingestion.py,sha256=GZkrke5_JJfA_PGOFc6ekbHKujHgMgqr6t4vop5n_bg,11210
 mlrun/feature_store/steps.py,sha256=jleXgtUbR_BDk0Q060vHYwfBwX01VoU3Y1s5S6QF8mo,29028
 mlrun/feature_store/retrieval/__init__.py,sha256=bwA4copPpLQi8fyoUAYtOyrlw0-6f3-Knct8GbJSvRg,1282
 mlrun/feature_store/retrieval/base.py,sha256=gzY7sesiy2sMQa6VtG2nSpscEIjcrsLm5QgS8LxM2nw,30732
 mlrun/feature_store/retrieval/dask_merger.py,sha256=Lxoj16c8hp9i5RHf-8BmG9NmReeEev6AHu96oEN2Gp8,4895
-mlrun/feature_store/retrieval/job.py,sha256=Amprb-iBaNUsnF2cf03JeALrGKCZsfZBDqSaeatnB9k,8007
+mlrun/feature_store/retrieval/job.py,sha256=CapLTnlN-5hzFVnC2wVXX1ydhXs_szNPKrAoOLK3cZc,8010
 mlrun/feature_store/retrieval/local_merger.py,sha256=M0R2FWc-kuLVvyAYbawrxAJvqjshtaRK2gQqF6DjgxM,4218
 mlrun/feature_store/retrieval/spark_merger.py,sha256=6A1YJF2W3zbMqnQ1WrBVzRpNExW9vX54ybREO2B9aXU,9819
 mlrun/feature_store/retrieval/storey_merger.py,sha256=5YM0UPrLjGOobulHkowRO-1LuvFD2cm_0GxcpnTdu0I,6314
@@ -368,7 +368,7 @@ mlrun/launcher/remote.py,sha256=neqIWNi64uR694DIFg6BdJk6Bbi0w_FIhfjChX1wsmk,6772
 mlrun/model_monitoring/__init__.py,sha256=XaYyvWsIXpjJQ2gCPj8tFvfSbRSEEqgDtNz4tCE5H4g,915
 mlrun/model_monitoring/api.py,sha256=MyS6dmFv2w9X-kGEPV4Z-Sn2Sihx85hmSe8Yd-NRQgI,34624
 mlrun/model_monitoring/application.py,sha256=OlpzGxBkTuBbqFIwN9AwiibqDP5YZaBHv6a7bWahm5c,12209
-mlrun/model_monitoring/batch.py,sha256=tnhTUkY9nDedkK7NZ5U53jL0P9i83_7QZlwEh1U0AhM,42337
+mlrun/model_monitoring/batch.py,sha256=kcSXUFaUp4KEiX_jihXvqjJW-IZmV56obEYpVrMwBFc,42654
 mlrun/model_monitoring/batch_application.py,sha256=kFGZUss7CBkpcei9NlHFmKoU1Wu5l7_pdq8GuOhK0bk,21938
 mlrun/model_monitoring/batch_application_handler.py,sha256=6I3XmganxCfI-IUFEvFpdsUFiw3OiIMq58BS-XARIMs,1046
 mlrun/model_monitoring/evidently_application.py,sha256=ToQ9BfXZJqhVSxsVyWVEpYjKYaaTAT9wB83Tj7KVJdA,3401
@@ -409,8 +409,8 @@ mlrun/platforms/iguazio.py,sha256=LU1d33ll5EKIyp2zitCffZIbq-3fRwNSNO9MK2cIsHc,21
 mlrun/platforms/other.py,sha256=z4pWqxXkVVuMLk-MbNb0Y_ZR5pmIsUm0R8vHnqpEnew,11852
 mlrun/projects/__init__.py,sha256=Lv5rfxyXJrw6WGOWJKhBz66M6t3_zsNMCfUD6waPwx4,1153
 mlrun/projects/operations.py,sha256=AJsin0LrJHTisAPi-9t2ciFNZ83QdMRESIG31Mh_0x4,17948
-mlrun/projects/pipelines.py,sha256=PJ9qgQzD3CsJojMVA7rtMTBRIHhcygBV9fLeYUjNKTI,36545
-mlrun/projects/project.py,sha256=7OWnIn5GWY6GsF2CrQ6SQG_IpyKccw4xwJPb1r7OwGs,138028
+mlrun/projects/pipelines.py,sha256=1ld4bzpUsqSmElwMBvEPdFJlDFenbGRhOok0BB4h1ug,36544
+mlrun/projects/project.py,sha256=P11VYOYntNZgUE6jxwULoT5N_VhJ3O-vcP8_BrcF0LY,138181
 mlrun/runtimes/__init__.py,sha256=OuwnzCoaoXnqAv_RhoYRF6mRPfZ2pkslenxASpXhHQM,6707
 mlrun/runtimes/base.py,sha256=qMOhtgnym5SlX9xXIGnC-KjwGng8a4yRI7KH1mqVC9U,38261
 mlrun/runtimes/constants.py,sha256=TnkD0nQ7pcFq1aJvbweoM4BgbkhaA6fHhg9rnAkxDBE,6689
@@ -470,11 +470,11 @@ mlrun/utils/notifications/notification/ipython.py,sha256=qrBmtECiRG6sZpCIVMg7RZc
 mlrun/utils/notifications/notification/slack.py,sha256=5JysqIpUYUZKXPSeeZtbl7qb2L9dj7p2NvnEBcEsZkA,3898
 mlrun/utils/notifications/notification/webhook.py,sha256=QHezCuN5uXkLcroAGxGrhGHaxAdUvkDLIsp27_Yrfd4,2390
 mlrun/utils/version/__init__.py,sha256=7kkrB7hEZ3cLXoWj1kPoDwo4MaswsI2JVOBpbKgPAgc,614
-mlrun/utils/version/version.json,sha256=jT2PVpQTitajCVEVdwH44HqEMoYdKzZFrwI4bekF9hU,89
+mlrun/utils/version/version.json,sha256=DVu8iNZNdV07ZdarkvN82a5XXbX_rb2pa6VDs5Lru-U,89
 mlrun/utils/version/version.py,sha256=HMwseV8xjTQ__6T6yUWojx_z6yUj7Io7O4NcCCH_sz8,1970
-mlrun-1.5.0rc16.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mlrun-1.5.0rc16.dist-info/METADATA,sha256=3Yg1j-Ur4F-nFSEuZ3wzjHm86iAwJwcme7ekaB6jQT0,17826
-mlrun-1.5.0rc16.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-mlrun-1.5.0rc16.dist-info/entry_points.txt,sha256=ZbXmb36B9JmK7EaleP8MIAbZSOQXQV0iwKR6si0HUWk,47
-mlrun-1.5.0rc16.dist-info/top_level.txt,sha256=NObLzw3maSF9wVrgSeYBv-fgnHkAJ1kEkh12DLdd5KM,6
-mlrun-1.5.0rc16.dist-info/RECORD,,
+mlrun-1.5.0rc17.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mlrun-1.5.0rc17.dist-info/METADATA,sha256=KKrbNsNaRqjfgMhKOqgvGzCElfKZPcK6VCwfI8U5Tfs,17826
+mlrun-1.5.0rc17.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+mlrun-1.5.0rc17.dist-info/entry_points.txt,sha256=ZbXmb36B9JmK7EaleP8MIAbZSOQXQV0iwKR6si0HUWk,47
+mlrun-1.5.0rc17.dist-info/top_level.txt,sha256=NObLzw3maSF9wVrgSeYBv-fgnHkAJ1kEkh12DLdd5KM,6
+mlrun-1.5.0rc17.dist-info/RECORD,,