Fix build GitHub action

Author: Florijan Stamenković

.github/workflows/python-package.yml

@@ -27,9 +27,9 @@ jobs:
     - name: set environment for branch
       run: |
           if [[ "${{github.base_ref}}" == "master" || "${{github.ref}}" == "refs/heads/master" ]]; then
-            echo "::set-env name=LABELBOX_TEST_ENVIRON::prod"
+            echo "LABELBOX_TEST_ENVIRON=prod" >> $GITHUB_ENV
           else
-            echo "::set-env name=LABELBOX_TEST_ENVIRON::staging"
+            echo "LABELBOX_TEST_ENVIRON=staging" >> $GITHUB_ENV
           fi
 
     - uses: actions/checkout@v2

labelbox/schema/project.py

@@ -367,7 +367,7 @@ def create_prediction(self, label, data_row, prediction_model=None):
         res = self.client.execute(query_str, params)
         return Prediction(self.client, res["createPrediction"])
 
-    def enable_model_assisted_labeling(self, toggle: bool=True) -> bool:
+    def enable_model_assisted_labeling(self, toggle: bool = True) -> bool:
         """ Turns model assisted labeling either on or off based on input
 
         Args:
@@ -385,15 +385,13 @@ def enable_model_assisted_labeling(self, toggle: bool=True) -> bool:
                     id, showingPredictionsToLabelers
                 }
             }
-        }""" % (project_param, show_param,project_param, show_param)
+        }""" % (project_param, show_param, project_param, show_param)
 
-        params = {
-            project_param: self.uid, 
-            show_param: toggle
-        }
+        params = {project_param: self.uid, show_param: toggle}
 
         res = self.client.execute(query_str, params)
-        return res["project"]["showPredictionsToLabelers"]["showingPredictionsToLabelers"]
+        return res["project"]["showPredictionsToLabelers"][
+            "showingPredictionsToLabelers"]
 
     def upload_annotations(
         self,
@@ -460,6 +458,7 @@ def _is_url_valid(url: Union[str, Path]) -> bool:
             raise ValueError(
                 f'Invalid annotations given of type: {type(annotations)}')
 
+
 class LabelingParameterOverride(DbObject):
     priority = Field.Int("priority")
     number_of_labels = Field.Int("number_of_labels")

tests/integration/test_toggle_mal.py

@@ -2,12 +2,13 @@
 from labelbox import Project
 import pytest
 
+
 def test_enable_model_assisted_labeling(project):
     response = project.enable_model_assisted_labeling()
     assert response == True
 
     response = project.enable_model_assisted_labeling(True)
     assert response == True
-    
+
     response = project.enable_model_assisted_labeling(False)
-    assert response == False
+    assert response == False