Cleanup schemanode cleanup code

Author: Val Brodsky

libs/labelbox/tests/conftest.py

@@ -25,21 +25,11 @@
 from labelbox.schema.queue_mode import QueueMode
 from labelbox import Client
 
-from labelbox import Dataset, DataRow
 from labelbox import LabelingFrontend
-from labelbox import OntologyBuilder, Tool, Option, Classification, MediaType
-from labelbox.orm import query
-from labelbox.pagination import PaginatedCollection
+from labelbox import OntologyBuilder, Tool, Option, Classification
 from labelbox.schema.annotation_import import LabelImport
-from labelbox.schema.catalog import Catalog
 from labelbox.schema.enums import AnnotationImportState
-from labelbox.schema.invite import Invite
-from labelbox.schema.quality_mode import QualityMode
-from labelbox.schema.queue_mode import QueueMode
-from labelbox.schema.user import User
 from labelbox.exceptions import LabelboxError
-from contextlib import suppress
-from labelbox import Client
 
 IMG_URL = "https://picsum.photos/200/300.jpg"
 MASKABLE_IMG_URL = "https://storage.googleapis.com/labelbox-datasets/image_sample_data/2560px-Kitano_Street_Kobe01s5s4110.jpeg"
@@ -734,7 +724,7 @@ def big_dataset(dataset: Dataset):
 
 @pytest.fixture
 def configured_batch_project_with_label(client, dataset, data_row,
-                                        wait_for_label_processing,
+                                        wait_for_label_processing, rand_gen,
                                         teardown_helpers):
     """Project with a batch having one datarow
     Project contains an ontology with 1 bbox tool
@@ -1164,23 +1154,9 @@ def teardown_project_labels_ontology_feature_schemas(project: Project):
         classification_feature_schema_ids = [
             feature["featureSchemaId"]
             for feature in ontology.normalized["classifications"]
-        ] + [
-            option["featureSchemaId"]
-            for feature in ontology.normalized["classifications"]
-            for option in feature.get("options", [])
         ]
-
         tool_feature_schema_ids = [
             feature["featureSchemaId"] for feature in ontology.normalized["tools"]
-        ] + [
-            classification["featureSchemaId"]
-            for tool in ontology.normalized["tools"]
-            for classification in tool.get("classifications", [])
-        ] + [
-            option["featureSchemaId"]
-            for tool in ontology.normalized["tools"]
-            for classification in tool.get("classifications", [])
-            for option in classification.get("options", [])
         ]
 
         feature_schema_ids = classification_feature_schema_ids + tool_feature_schema_ids