Change decorator

Author: dshabin

superannotate/analytics/class_analytics.py

@@ -8,10 +8,12 @@
 from plotly.subplots import make_subplots
 
 from .common import aggregate_annotations_as_df
-from ..mixp.decorators import trackable
+from ..mixp.decorators import Trackable
+
 logger = logging.getLogger("superannotate-python-sdk")
 
-@trackable
+
+@Trackable
 def class_distribution(export_root, project_names, visualize=False):
     """Aggregate distribution of classes across multiple projects.
 
@@ -63,7 +65,7 @@ def class_distribution(export_root, project_names, visualize=False):
     return df
 
 
-@trackable
+@Trackable
 def attribute_distribution(export_root, project_names, visualize=False):
     """Aggregate distribution of attributes across multiple projects.
 

superannotate/analytics/common.py

@@ -6,10 +6,12 @@
 import pandas as pd
 
 from ..exceptions import SABaseException
-from ..mixp.decorators import trackable
+from ..mixp.decorators import Trackable
+
 logger = logging.getLogger("superannotate-python-sdk")
 
-@trackable
+
+@Trackable
 def df_to_annotations(df, output_dir):
     """Converts and saves pandas DataFrame annotation info (see aggregate_annotations_as_df)
     in output_dir.
@@ -147,7 +149,8 @@ def df_to_annotations(df, output_dir):
         indent=4
     )
 
-@trackable
+
+@Trackable
 def aggregate_annotations_as_df(
     project_root,
     include_classes_wo_annotations=False,

superannotate/consensus_benchmark/benchmark.py

@@ -9,12 +9,12 @@
 from .helpers import image_consensus, consensus_plot
 from ..db.exports import prepare_export, download_export
 from ..analytics.common import aggregate_annotations_as_df
-from ..mixp.decorators import trackable
+from ..mixp.decorators import Trackable
 
 logger = logging.getLogger("superannotate-python-sdk")
 
 
-@trackable
+@Trackable
 def benchmark(
     project,
     gt_folder,

superannotate/consensus_benchmark/consensus.py

@@ -9,12 +9,12 @@
 from .helpers import image_consensus, consensus_plot
 from ..db.exports import prepare_export, download_export
 from ..analytics.common import aggregate_annotations_as_df
-from ..mixp.decorators import trackable
+from ..mixp.decorators import Trackable
 
 logger = logging.getLogger("superannotate-python-sdk")
 
 
-@trackable
+@Trackable
 def consensus(
     project,
     folder_names,

superannotate/dataframe_filtering.py

@@ -1,7 +1,8 @@
 import pandas as pd
-from .mixp.decorators import trackable
+from .mixp.decorators import Trackable
 
-@trackable
+
+@Trackable
 def filter_images_by_comments(
     annotations_df,
     include_unresolved_comments=True,
@@ -40,7 +41,8 @@ def filter_images_by_comments(
 
     return list(images)
 
-@trackable
+
+@Trackable
 def filter_images_by_tags(annotations_df, include=None, exclude=None):
     """Filter images on tags
 
@@ -74,7 +76,8 @@ def filter_images_by_tags(annotations_df, include=None, exclude=None):
 
     return list(images)
 
-@trackable
+
+@Trackable
 def filter_annotation_instances(annotations_df, include=None, exclude=None):
     """Filter annotation instances from project annotations pandas DataFrame.
 

superannotate/db/annotation_classes.py

@@ -11,12 +11,14 @@
     SANonExistingAnnotationClassNameException
 )
 from .project_api import get_project_metadata_bare
-from ..mixp.decorators import trackable
+from ..mixp.decorators import Trackable
+
 logger = logging.getLogger("superannotate-python-sdk")
 
 _api = API.get_instance()
 
-@trackable
+
+@Trackable
 def create_annotation_class(project, name, color, attribute_groups=None):
     """Create annotation class in project
 
@@ -78,7 +80,8 @@ def create_annotation_class(project, name, color, attribute_groups=None):
     new_class = res[0]
     return new_class
 
-@trackable
+
+@Trackable
 def delete_annotation_class(project, annotation_class):
     """Deletes annotation class from project
 
@@ -112,7 +115,8 @@ def delete_annotation_class(project, annotation_class):
             "Couldn't delete annotation class " + response.text
         )
 
-@trackable
+
+@Trackable
 def create_annotation_classes_from_classes_json(
     project, classes_json, from_s3_bucket=None
 ):
@@ -199,7 +203,8 @@ def del_unn(d):
     assert len(res) == len(new_classes)
     return res
 
-@trackable
+
+@Trackable
 def search_annotation_classes(project, name_prefix=None):
     """Searches annotation classes by name_prefix (case-insensitive)
 
@@ -238,7 +243,8 @@ def search_annotation_classes(project, name_prefix=None):
 
     return result_list
 
-@trackable
+
+@Trackable
 def get_annotation_class_metadata(project, annotation_class_name):
     """Returns annotation class metadata
 
@@ -271,7 +277,8 @@ def get_annotation_class_metadata(project, annotation_class_name):
             " doesn't exist."
         )
 
-@trackable
+
+@Trackable
 def download_annotation_classes_json(project, folder):
     """Downloads project classes.json to folder
 

superannotate/db/exports.py

@@ -19,7 +19,7 @@
     SANonExistingExportNameException
 )
 from .project_api import get_project_metadata_bare
-from ..mixp.decorators import trackable
+from ..mixp.decorators import Trackable
 
 logger = logging.getLogger("superannotate-python-sdk")
 
@@ -57,7 +57,7 @@ def get_export_metadata(project, export_name):
         )
 
 
-@trackable
+@Trackable
 def get_exports(project, return_metadata=False):
     """Get all prepared exports of the project.
 
@@ -99,7 +99,7 @@ def _get_export(export):
     return response.json()
 
 
-@trackable
+@Trackable
 def prepare_export(
     project,
     folder_names=None,
@@ -222,7 +222,7 @@ def _download_file(url, local_filename):
     return local_filename
 
 
-@trackable
+@Trackable
 def download_export(
     project, export, folder_path, extract_zip_contents=True, to_s3_bucket=None
 ):