style check

capitalone · Jul 18, 2024 · 4e5dbce · 4e5dbce
1 parent 566be50
commit 4e5dbce
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 9 deletions.
diff --git a/rubicon_ml/intake_rubicon/__init__.py b/rubicon_ml/intake_rubicon/__init__.py
@@ -1,6 +1,13 @@
 import intake  # noqa F401
 
 from rubicon_ml.intake_rubicon.experiment import ExperimentSource
-from rubicon_ml.intake_rubicon.viz import ExperimentsTableDataSource, MetricCorrelationPlotDataSource
+from rubicon_ml.intake_rubicon.viz import (
+    ExperimentsTableDataSource,
+    MetricCorrelationPlotDataSource,
+)
 
-__all__ = ["ExperimentSource", "ExperimentsTableDataSource", "MetricCorrelationPlotDataSource",]
+__all__ = [
+    "ExperimentSource",
+    "ExperimentsTableDataSource",
+    "MetricCorrelationPlotDataSource",
+]
diff --git a/rubicon_ml/intake_rubicon/publish.py b/rubicon_ml/intake_rubicon/publish.py
@@ -6,7 +6,6 @@
 if TYPE_CHECKING:
     from rubicon_ml.viz.experiments_table import ExperimentsTable
     from rubicon_ml.viz.metric_correlation_plot import MetricCorrelationPlot
-
 
 
 def publish(
@@ -105,7 +104,7 @@ def _update_catalog(
 def _build_catalog(experiments, visualization):
     from rubicon_ml.viz.experiments_table import ExperimentsTable
     from rubicon_ml.viz.metric_correlation_plot import MetricCorrelationPlot
-    
+
     """Helper function to build catalog dictionary from given experiments.
 
     Parameters
@@ -162,5 +161,5 @@ def _build_catalog(experiments, visualization):
             catalog["sources"]["metric_correlation_plot"] = appended_visualization_catalog
 
         # append visualization object to end of catalog file
-        
+
     return catalog
diff --git a/rubicon_ml/intake_rubicon/viz.py b/rubicon_ml/intake_rubicon/viz.py
@@ -23,7 +23,8 @@ def _get_schema(self):
         self._visualization_object = ExperimentsTable(**self._catalog_data)
 
         return super()._get_schema()
-
+
+
 class MetricCorrelationPlotDataSource(VizDataSourceMixin):
     """An Intake data source for reading `rubicon` Metric Correlation Plot visualizations."""
 

diff --git a/tests/unit/intake_rubicon/test_publish.py b/tests/unit/intake_rubicon/test_publish.py
@@ -7,7 +7,6 @@
 
 
 def test_publish(project_client):
-
     # Experiment Table
 
     project = project_client

diff --git a/tests/unit/intake_rubicon/test_viz.py b/tests/unit/intake_rubicon/test_viz.py
@@ -1,6 +1,9 @@
 import os
 
-from rubicon_ml.intake_rubicon.viz import ExperimentsTableDataSource, MetricCorrelationPlotDataSource
+from rubicon_ml.intake_rubicon.viz import (
+    ExperimentsTableDataSource,
+    MetricCorrelationPlotDataSource,
+)
 
 root = os.path.dirname(__file__)
 
@@ -34,6 +37,7 @@ def test_experiments_table_source():
 
     source.close()
 
+
 def test_metric_correlation_plot_source():
     catalog_data_sample = {
         "metric_names": None,
@@ -54,4 +58,3 @@ def test_metric_correlation_plot_source():
     assert visualization.selected_metric == catalog_data_sample["selected_metric"]
 
     source.close()
-
Original file line number	Diff line number	Diff line change
Expand Up		@@ -7,7 +7,6 @@


		def test_publish(project_client):

		# Experiment Table

		project = project_client
Expand Down