yapf

Author: Matt Sokoloff

labelbox/schema/task.py

@@ -40,7 +40,6 @@ class Task(DbObject):
     created_by = Relationship.ToOne("User", False, "created_by")
     organization = Relationship.ToOne("Organization")
 
-
     def refresh(self) -> None:
         """ Refreshes Task data from the server. """
         assert self._user is not None
@@ -74,7 +73,7 @@ def wait_till_done(self, timeout_seconds=300) -> None:
     def errors(self) -> Optional[Dict[str, Any]]:
         """ Downloads the result file from Task
         """
-        self.wait_till_done(timeout_seconds = 600)
+        self.wait_till_done(timeout_seconds=600)
         if self.status == "FAILED":
             data = self._fetch_remote(self.result_url)
             if data:
@@ -87,7 +86,7 @@ def errors(self) -> Optional[Dict[str, Any]]:
     def result(self) -> Dict[str, Any]:
         """ Fetch the result for a task
         """
-        self.wait_till_done(timeout_seconds = 600)
+        self.wait_till_done(timeout_seconds=600)
         if self.status == "COMPLETE":
             return self._fetch_remote(self.result_url)
         elif self.status == "FAILED":
@@ -102,4 +101,3 @@ def _fetch_remote(self, result_url) -> Dict[str, Any]:
         response = requests.get(result_url)
         response.raise_for_status()
         return response.json()
-

tests/data/metrics/confusion_matrix/t.py

@@ -0,0 +1,23 @@
+import json
+
+chain = dict()
+t = 3
+for a0 in range(t):
+    nn = 4
+    answer = dict()
+    for n in range(1, nn + 1):
+        vals = []
+        offset = 0
+        vals.append(n)  #[2, 1]
+        while (not n == 1):
+            if (n % 2):
+                n = 3 * n + 1
+            else:
+                n //= 2
+            vals.append(n)
+        for index, val in enumerate(vals):
+            chain[val] = len(vals) - index + offset - 1
+
+        print(n)
+        answer[n] = chain[n]
+    print(json.dumps(answer, sort_keys=True))

tests/integration/test_data_rows.py

@@ -547,18 +547,17 @@ def test_delete_data_row_attachment(datarow, image_url):
 
 def test_create_data_rows_result(client, dataset, image_url):
     task = dataset.create_data_rows([
-            {
-                DataRow.row_data: image_url,
-                DataRow.external_id: "row1",
-            },
-            {
-                DataRow.row_data: image_url,
-                DataRow.external_id: "row1",
-            },
-        ])
+        {
+            DataRow.row_data: image_url,
+            DataRow.external_id: "row1",
+        },
+        {
+            DataRow.row_data: image_url,
+            DataRow.external_id: "row1",
+        },
+    ])
     assert task.errors is None
     assert 'createdDataRows' in task.result
     results = task.result['createdDataRows']
     for result in results:
         client.get_data_row(result['id'])
-

tests/integration/test_task.py

@@ -38,4 +38,3 @@ def test_task_success(dataset, image_url):
     assert task.status == "COMPLETE"
     assert task.errors is None
     assert task.result is not None
-