Add build job invalidation apis

Author: mmourafiq

polyaxon_client/api/build_job.py

@@ -152,6 +152,28 @@ def create_code_reference(self,
                 e=e, log_message='Error while creating build coderef.')
             return None
 
+    def invalidate(self,
+                   username,
+                   project_name,
+                   job_id,
+                   background=False):
+        request_url = self.build_url(self._get_http_url(),
+                                     username,
+                                     project_name,
+                                     'builds',
+                                     job_id,
+                                     'invalidate')
+        if background:
+            self.transport.async_post(request_url)
+            return None
+
+        try:
+            return self.transport.post(request_url)
+        except PolyaxonClientException as e:
+            self.transport.handle_exception(
+                e=e, log_message='Error while invalidating build.')
+            return None
+
     def send_logs(self,
                   username,
                   project_name,

polyaxon_client/api/project.py

@@ -359,6 +359,22 @@ def list_builds(self, username, project_name, query=None, sort=None, page=1):
             self.transport.handle_exception(e=e, log_message='Error while retrieving build jobs.')
             return []
 
+    def invalidate_builds(self, username, project_name, background=False):
+        """Invalidate all build jobs related to this project."""
+        request_url = self.build_url(
+            self._get_http_url(), username, project_name, 'builds', 'invalidate')
+
+        if background:
+            self.transport.async_post(request_url)
+            return None
+
+        try:
+            return self.transport.post(request_url)
+        except PolyaxonClientException as e:
+            self.transport.handle_exception(
+                e=e, log_message='Error while invalidating builds.')
+            return None
+
     def create_build(self, username, project_name, build_config, background=False):
         build_config = self.validate_config(config=build_config, config_schema=BuildJobConfig)
         request_url = self.build_url(self._get_http_url(), username, project_name, 'builds')

polyaxon_client/tracking/contrib/fastai.py

@@ -1,9 +1,9 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, division, print_function
 
+from polyaxon_client import settings
 from polyaxon_client.exceptions import PolyaxonClientException
 from polyaxon_client.tracking import Experiment
-from polyaxon_client import settings
 
 try:
     from fastai.callbacks import TrackerCallback

polyaxon_client/tracking/contrib/keras.py

@@ -1,9 +1,9 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, division, print_function
 
+from polyaxon_client import settings
 from polyaxon_client.exceptions import PolyaxonClientException
 from polyaxon_client.tracking import Experiment
-from polyaxon_client import settings
 
 try:
     from keras.callbacks import Callback

polyaxon_client/tracking/contrib/logging_tensor_hook.py

@@ -1,9 +1,9 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import, division, print_function
 
+from polyaxon_client import settings
 from polyaxon_client.exceptions import PolyaxonClientException
 from polyaxon_client.tracking import Experiment
-from polyaxon_client import settings
 
 try:
     from tensorflow.train import LoggingTensorHook

tests/test_api/test_build_job.py

@@ -242,6 +242,29 @@ def test_stop_build(self):
                 'username', 'project_name', 1, background=True),
             method='post')
 
+    @httpretty.activate
+    def test_invalidate_build(self):
+        httpretty.register_uri(
+            httpretty.POST,
+            BaseApiHandler.build_url(
+                self.api_config.base_url,
+                '/',
+                'username',
+                'project_name',
+                'builds',
+                1,
+                'invalidate'),
+            content_type='application/json',
+            status=200)
+        result = self.api_handler.invalidate('username', 'project_name', 1)
+        assert result.status_code == 200
+
+        # Async
+        self.assert_async_call(
+            api_handler_call=lambda: self.api_handler.invalidate(
+                'username', 'project_name', 1, background=True),
+            method='post')
+
     @httpretty.activate
     def test_bookmark_build(self):
         httpretty.register_uri(

tests/test_api/test_project.py

@@ -974,6 +974,28 @@ def test_create_build(self):
                 'user', 'project', obj.to_dict(), background=True),
             method='post')
 
+    @httpretty.activate
+    def test_invalidate_builds(self):
+        httpretty.register_uri(
+            httpretty.POST,
+            BaseApiHandler.build_url(
+                self.api_config.base_url,
+                '/',
+                'username',
+                'project_name',
+                'builds',
+                'invalidate'),
+            content_type='application/json',
+            status=200)
+        result = self.api_handler.invalidate_builds('username', 'project_name')
+        assert result.status_code == 200
+
+        # Async
+        self.assert_async_call(
+            api_handler_call=lambda: self.api_handler.invalidate_builds(
+                'username', 'project_name', background=True),
+            method='post')
+
     @httpretty.activate
     def test_start_tensorboard(self):
         httpretty.register_uri(