Lint

haupt/haupt/apis/bookmarks/views.py

@@ -38,7 +38,7 @@ def destroy(self, request, *args, **kwargs):
             Models.Bookmark,
             **user_filter,
             content_type__model=self.content_type,
-            object_id=obj.id
+            object_id=obj.id,
         )
         bookmark.enabled = False
         bookmark.save(update_fields=["enabled"])

haupt/haupt/apis/runs/views.py

@@ -118,7 +118,7 @@ def clone(self, obj, content, **kwargs):
             description=kwargs.get("description"),
             tags=kwargs.get("tags"),
             meta_info=kwargs.get("meta_info"),
-            **self._get_additional_fields(obj)
+            **self._get_additional_fields(obj),
         )
 
 
@@ -138,7 +138,7 @@ def clone(self, obj, content, **kwargs):
             tags=kwargs.get("tags"),
             meta_info=kwargs.get("meta_info"),
             message="Run was resumed by user.",
-            **self._get_additional_fields(obj)
+            **self._get_additional_fields(obj),
         )
 
     def pre_validate(self, obj):

haupt/haupt/cli/sandbox.py

@@ -1,5 +1,3 @@
-import os
-
 import click
 
 from polyaxon.logger import clean_outputs

haupt/haupt/cli/viewer.py

@@ -1,5 +1,3 @@
-import os
-
 from collections.abc import Mapping
 from typing import Dict, Optional
 

haupt/haupt/common/apis/urls/runs.py

@@ -5,10 +5,6 @@
     RUN_UUID_PATTERN,
 )
 
-URLS_RUNS_CREATE = r"^{}/{}/runs/create/?$".format(
-    OWNER_NAME_PATTERN, PROJECT_NAME_PATTERN
-)
-URLS_RUNS_LIST = r"^{}/runs/list/?$".format(OWNER_NAME_PATTERN, PROJECT_NAME_PATTERN)
 URLS_RUNS_DETAILS = r"^{}/{}/runs/{}/?$".format(
     OWNER_NAME_PATTERN, PROJECT_NAME_PATTERN, RUN_UUID_PATTERN
 )

haupt/haupt/common/authentication/internal.py

@@ -66,7 +66,9 @@ def authenticate(
         return self.authenticate_credentials(internal_service, token)
 
     def authenticate_credentials(
-        self, service: str, key: str  # pylint:disable=arguments-differ
+        self,
+        service: str,
+        key: str,  # pylint:disable=arguments-differ
     ) -> Optional[Tuple["InternalUser", None]]:
         internal_user = InternalUser(service=service)
         if internal_user.access_token != key:

haupt/haupt/common/endpoints/files.py

@@ -18,7 +18,7 @@ def __init__(self, *args, as_attachment=False, filepath="", **kwargs):
             as_attachment=as_attachment,
             filename=filename,
             headers=headers,
-            **kwargs
+            **kwargs,
         )
 
     def get_stat_headers(self, filepath: str) -> Optional[Dict]:

haupt/haupt/common/events/event.py

@@ -84,7 +84,7 @@ def __init__(
         instance_uuid: Optional[str] = None,
         ref_id: Optional[str] = None,
         event_data: Mapping = None,
-        **items
+        **items,
     ):
         self.uuid = UUID(uid) if uid else uuid1()
         self.datetime = datetime or timezone.now()
@@ -250,7 +250,7 @@ def from_event_data(cls, event_data: Mapping, **kwargs) -> "Event":
             instance=event_data.get("instance"),
             instance_id=event_data.get("instance_id"),
             instance_uuid=event_data.get("instance_uuid"),
-            **kwargs
+            **kwargs,
         )
 
 

haupt/haupt/common/events/event_service.py

@@ -17,7 +17,7 @@ def get_event(
         event_type: str,
         event_data: Mapping = None,
         instance: Any = None,
-        **kwargs
+        **kwargs,
     ) -> Event:
         if instance or not event_data:
             return self.event_manager.get(event_type).from_instance(instance, **kwargs)
@@ -30,7 +30,7 @@ def record(
         event_type: str,
         event_data: Mapping = None,
         instance: Any = None,
-        **kwargs
+        **kwargs,
     ) -> Optional[Event]:
         """Validate and record an event.
 

haupt/haupt/orchestration/executor/handlers.py

@@ -81,9 +81,7 @@ def handle_run_created(cls, workers_backend, event: "Event") -> None:  # noqa: F
         )
 
     @classmethod
-    def handle_run_approved_triggered(
-        cls, workers_backend, event: "Event"
-    ) -> None:  # noqa: F821
+    def handle_run_approved_triggered(cls, workers_backend, event: "Event") -> None:  # noqa: F821
         run = cls.MANAGER.get_run(run_id=event.instance_id, run=event.instance)
         if not run:
             return
@@ -101,21 +99,15 @@ def handle_run_approved_triggered(
         )
 
     @classmethod
-    def handle_run_stopped_triggered(
-        cls, workers_backend, event: "Event"
-    ) -> None:  # noqa: F821
+    def handle_run_stopped_triggered(cls, workers_backend, event: "Event") -> None:  # noqa: F821
         cls.MANAGER.runs_stop(run_id=event.instance_id, run=event.instance)
 
     @classmethod
-    def handle_run_skipped_triggered(
-        cls, workers_backend, event: "Event"
-    ) -> None:  # noqa: F821
+    def handle_run_skipped_triggered(cls, workers_backend, event: "Event") -> None:  # noqa: F821
         cls.MANAGER.runs_skip(run_id=event.instance_id, run=event.instance)
 
     @classmethod
-    def handle_run_done(
-        cls, workers_backend, event: "Event" = None
-    ) -> None:  # noqa: F821
+    def handle_run_done(cls, workers_backend, event: "Event" = None) -> None:  # noqa: F821
         """Handles all run done statuses"""
         workers_backend.send(
             SchedulerCeleryTasks.RUNS_NOTIFY_DONE,
@@ -125,9 +117,7 @@ def handle_run_done(
         )
 
     @classmethod
-    def handle_run_new_status(
-        cls, workers_backend, event: "Event"
-    ) -> None:  # noqa: F821
+    def handle_run_new_status(cls, workers_backend, event: "Event") -> None:  # noqa: F821
         run = cls.MANAGER.get_run(run_id=event.instance_id, run=event.instance)
         if not run:
             return
@@ -143,9 +133,7 @@ def handle_run_new_status(
             )
 
     @classmethod
-    def handle_new_artifacts(
-        cls, workers_backend, event: "Event"
-    ) -> None:  # noqa: F821
+    def handle_new_artifacts(cls, workers_backend, event: "Event") -> None:  # noqa: F821
         artifacts = event.data.get("artifacts")
         if not artifacts:
             return

haupt/haupt/orchestration/scheduler/resolver.py

@@ -1628,8 +1628,9 @@ def _get_project_version_refs(
         # }
         if len(ids) != len(names):
             raise AccessNotFound(
-                "Some {} version refs were provided "
-                "but were not found: `{}`".format(kind, names)
+                "Some {} version refs were provided but were not found: `{}`".format(
+                    kind, names
+                )
             )
 
         # if (

haupt/haupt/polyconf/asgi/viewer.py

@@ -6,6 +6,7 @@
 For more information on this file, see
 https://docs.djangoproject.com/en/4.1/howto/deployment/asgi/
 """
+
 import logging
 import os
 

haupt/haupt/polyconf/wsgi.py

@@ -6,6 +6,7 @@
 For more information on this file, see
 https://docs.djangoproject.com/en/1.7/howto/deployment/wsgi/
 """
+
 import os
 
 from django.core.wsgi import get_wsgi_application

haupt/tests/test_apis/test_project_resources/test_runs_views.py

@@ -1428,13 +1428,13 @@ def test_get_runs_for_original(self):
         assert resp.data["next"] is None
         assert resp.data["count"] == 2
 
-        resp = self.client.get(self.url + f"?query=cloning_kind:cache")
+        resp = self.client.get(self.url + "?query=cloning_kind:cache")
         assert resp.status_code == status.HTTP_200_OK
 
         assert resp.data["next"] is None
         assert resp.data["count"] == 1
 
-        resp = self.client.get(self.url + f"?query=cloning_kind:cache")
+        resp = self.client.get(self.url + "?query=cloning_kind:cache")
         assert resp.status_code == status.HTTP_200_OK
 
         assert resp.data["next"] is None

haupt/tests/test_apis/test_projects/test_views.py

@@ -1,7 +1,6 @@
 import pytest
 
 from flaky import flaky
-from mock.mock import patch
 
 from rest_framework import status
 

haupt/tests/test_apis/test_run_lineage/test_lineage_clones_views.py

@@ -115,17 +115,17 @@ def test_get_filter(self):  # pylint:disable=too-many-statements
         assert resp.data["count"] == 0
 
         # Values
-        resp = self.client.get(self.url + f"?query=cloning_kind:restart")
+        resp = self.client.get(self.url + "?query=cloning_kind:restart")
         assert resp.status_code == status.HTTP_200_OK
         assert resp.data["next"] is None
         assert resp.data["count"] == 1
 
-        resp = self.client.get(self.url + f"?query=cloning_kind:copy")
+        resp = self.client.get(self.url + "?query=cloning_kind:copy")
         assert resp.status_code == status.HTTP_200_OK
         assert resp.data["next"] is None
         assert resp.data["count"] == 1
 
-        resp = self.client.get(self.url + f"?query=cloning_kind:cache")
+        resp = self.client.get(self.url + "?query=cloning_kind:cache")
         assert resp.status_code == status.HTTP_200_OK
         assert resp.data["next"] is None
         assert resp.data["count"] == 1

haupt/tests/test_apis/test_run_lineage/test_lineage_edges_views.py

@@ -264,12 +264,12 @@ def test_get_filter(self):  # pylint:disable=too-many-statements
         assert resp.data["count"] == 0
 
         # Values
-        resp = self.client.get(self.url + f"?query=values.foo:bar")
+        resp = self.client.get(self.url + "?query=values.foo:bar")
         assert resp.status_code == status.HTTP_200_OK
         assert resp.data["next"] is None
         assert resp.data["count"] == 4
 
-        resp = self.client.get(self.url + f"?query=values.foo:moo")
+        resp = self.client.get(self.url + "?query=values.foo:moo")
         assert resp.status_code == status.HTTP_200_OK
         assert resp.data["next"] is None
         assert resp.data["count"] == 0