From b0e2ed538f6e9bb629864cba6c852932b860bc6b Mon Sep 17 00:00:00 2001 From: Brian Wylie Date: Mon, 25 Dec 2023 17:07:57 -0700 Subject: [PATCH] linter formatting --- src/sageworks/utils/sageworks_logging.py | 1 - tests/create_basic_test_artifacts.py | 13 +++++++------ 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/src/sageworks/utils/sageworks_logging.py b/src/sageworks/utils/sageworks_logging.py index 56a1857fe..b7afd7f9e 100644 --- a/src/sageworks/utils/sageworks_logging.py +++ b/src/sageworks/utils/sageworks_logging.py @@ -12,7 +12,6 @@ def __init__(self, rate_seconds=60): self.last_log_times = defaultdict(lambda: 0) def filter(self, record): - # Get the message and last log time for this message message = str(record.msg) last_log_time = self.last_log_times[message] diff --git a/tests/create_basic_test_artifacts.py b/tests/create_basic_test_artifacts.py index 4887f1645..66fada447 100644 --- a/tests/create_basic_test_artifacts.py +++ b/tests/create_basic_test_artifacts.py @@ -37,7 +37,6 @@ # Create the test_data DataSource if recreate or not DataSource("test_data").exists(): - # Create a new Data Source from a dataframe of test data test_data = TestDataGenerator() df = test_data.person_data() @@ -60,11 +59,13 @@ # Create the abalone_regression Model if recreate or not Model("abalone-regression").exists(): fs = FeatureSet("abalone_features") - fs.to_model(ModelType.REGRESSOR, - name="abalone-regression", - target_column="class_number_of_rings", - tags=["abalone", "regression"], - description="Abalone Regression Model") + fs.to_model( + ModelType.REGRESSOR, + name="abalone-regression", + target_column="class_number_of_rings", + tags=["abalone", "regression"], + description="Abalone Regression Model", + ) # Create the abalone_regression Endpoint if recreate or not Endpoint("abalone-regression-end").exists():