fix bug #76

giotto-ai · Jun 2, 2022 · b3c91a3 · b3c91a3
1 parent bb0c449
commit b3c91a3
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 6 deletions.
diff --git a/gdeep/search/__init__.py b/gdeep/search/__init__.py
@@ -11,7 +11,6 @@
     'HPOConfig',
     'HyperParameterOptimization',
     'GiottoSummaryWriter',
-    '_benchmarking_param',
     'PersformerHyperparameterSearch',
     'clean_up_files'
     ]
diff --git a/gdeep/utility/__init__.py b/gdeep/utility/__init__.py
@@ -1,6 +1,6 @@
 from .utils import _are_compatible, save_model_and_optimizer, \
     ensemble_wrapper, _inner_refactor_scalars, is_notebook, \
-        autoreload_if_notebook, KnownWarningSilencer
+    autoreload_if_notebook, KnownWarningSilencer
 from .constants import ROOT_DIR, DEFAULT_DATA_DIR, DATASET_BUCKET_NAME, \
     DEFAULT_DOWNLOAD_DIR, DATASET_BUCKET_NAME, DEFAULT_GRAPH_DIR, \
     DEVICE
@@ -9,12 +9,9 @@
     get_return_type
 
 __all__ = [
-    '_are_compatible',
     'save_model_and_optimizer',
     'optimisation',
     'ensemble_wrapper',
-    'intersection_homology',
-    '_inner_refactor_scalars',  # This should be here
     'is_notebook',
     'autoreload_if_notebook',
     'ROOT_DIR',
@@ -28,4 +25,4 @@
     'torch_transform',
     'get_parameter_types',
     'KnownWarningSilencer'
-    ]
+    ]
diff --git a/gdeep/visualisation/tests/__init__.py b/gdeep/visualisation/tests/__init__.py