mlrun 1.3.3rc1__py3-none-any.whl → 1.4.0__py3-none-any.whl

mlrun/api/api/endpoints/background_tasks.py

@@ -1,4 +1,4 @@
-# Copyright 2018 Iguazio
+# Copyright 2023 Iguazio
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -18,10 +18,10 @@ import sqlalchemy.orm
 from fastapi.concurrency import run_in_threadpool
 
 import mlrun.api.api.deps
-import mlrun.api.schemas
 import mlrun.api.utils.auth.verifier
 import mlrun.api.utils.background_tasks
 import mlrun.api.utils.clients.chief
+import mlrun.common.schemas
 from mlrun.utils import logger
 
 router = fastapi.APIRouter()
@@ -29,12 +29,12 @@ router = fastapi.APIRouter()
 
 @router.get(
     "/projects/{project}/background-tasks/{name}",
-    response_model=mlrun.api.schemas.BackgroundTask,
+    response_model=mlrun.common.schemas.BackgroundTask,
 )
 async def get_project_background_task(
     project: str,
     name: str,
-    auth_info: mlrun.api.schemas.AuthInfo = fastapi.Depends(
+    auth_info: mlrun.common.schemas.AuthInfo = fastapi.Depends(
         mlrun.api.api.deps.authenticate_request
     ),
     db_session: sqlalchemy.orm.Session = fastapi.Depends(
@@ -44,10 +44,10 @@ async def get_project_background_task(
     # Since there's no not-found option on get_project_background_task - we authorize before getting (unlike other
     # get endpoint)
     await mlrun.api.utils.auth.verifier.AuthVerifier().query_project_resource_permissions(
-        mlrun.api.schemas.AuthorizationResourceTypes.project_background_task,
+        mlrun.common.schemas.AuthorizationResourceTypes.project_background_task,
         project,
         name,
-        mlrun.api.schemas.AuthorizationAction.read,
+        mlrun.common.schemas.AuthorizationAction.read,
         auth_info,
     )
     return await run_in_threadpool(
@@ -60,12 +60,12 @@ async def get_project_background_task(
 
 @router.get(
     "/background-tasks/{name}",
-    response_model=mlrun.api.schemas.BackgroundTask,
+    response_model=mlrun.common.schemas.BackgroundTask,
 )
 async def get_internal_background_task(
     name: str,
     request: fastapi.Request,
-    auth_info: mlrun.api.schemas.AuthInfo = fastapi.Depends(
+    auth_info: mlrun.common.schemas.AuthInfo = fastapi.Depends(
         mlrun.api.api.deps.authenticate_request
     ),
 ):
@@ -76,14 +76,14 @@ async def get_internal_background_task(
     igz_version = mlrun.mlconf.get_parsed_igz_version()
     if igz_version and igz_version >= semver.VersionInfo.parse("3.7.0-b1"):
         await mlrun.api.utils.auth.verifier.AuthVerifier().query_resource_permissions(
-            mlrun.api.schemas.AuthorizationResourceTypes.background_task,
+            mlrun.common.schemas.AuthorizationResourceTypes.background_task,
             name,
-            mlrun.api.schemas.AuthorizationAction.read,
+            mlrun.common.schemas.AuthorizationAction.read,
             auth_info,
         )
     if (
         mlrun.mlconf.httpdb.clusterization.role
-        != mlrun.api.schemas.ClusterizationRole.chief
+        != mlrun.common.schemas.ClusterizationRole.chief
     ):
         logger.info(
             "Requesting internal background task, re-routing to chief",

mlrun/api/api/endpoints/client_spec.py

@@ -1,4 +1,4 @@
-# Copyright 2018 Iguazio
+# Copyright 2023 Iguazio
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -17,21 +17,21 @@ import typing
 from fastapi import APIRouter, Header
 
 import mlrun.api.crud
-import mlrun.api.schemas
+import mlrun.common.schemas
 
 router = APIRouter()
 
 
 @router.get(
     "/client-spec",
-    response_model=mlrun.api.schemas.ClientSpec,
+    response_model=mlrun.common.schemas.ClientSpec,
 )
 def get_client_spec(
     client_version: typing.Optional[str] = Header(
-        None, alias=mlrun.api.schemas.HeaderNames.client_version
+        None, alias=mlrun.common.schemas.HeaderNames.client_version
     ),
     client_python_version: typing.Optional[str] = Header(
-        None, alias=mlrun.api.schemas.HeaderNames.python_version
+        None, alias=mlrun.common.schemas.HeaderNames.python_version
     ),
 ):
     return mlrun.api.crud.ClientSpec().get_client_spec(

mlrun/api/api/endpoints/clusterization_spec.py

@@ -1,4 +1,4 @@
-# Copyright 2018 Iguazio
+# Copyright 2023 Iguazio
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -15,17 +15,19 @@
 import fastapi
 
 import mlrun.api.crud
-import mlrun.api.schemas
 import mlrun.api.utils.clients.chief
+import mlrun.common.schemas
 
 router = fastapi.APIRouter()
 
 
-@router.get("/clusterization-spec", response_model=mlrun.api.schemas.ClusterizationSpec)
+@router.get(
+    "/clusterization-spec", response_model=mlrun.common.schemas.ClusterizationSpec
+)
 async def clusterization_spec():
     if (
         mlrun.mlconf.httpdb.clusterization.role
-        != mlrun.api.schemas.ClusterizationRole.chief
+        != mlrun.common.schemas.ClusterizationRole.chief
     ):
         chief_client = mlrun.api.utils.clients.chief.Client()
         return await chief_client.get_clusterization_spec()