error found, removed print statements

src/data_generator.py

@@ -98,7 +98,6 @@ class DataGenerator(keras.utils.Sequence):
         station = self.get_station_key(key)
         data = DataPrep(self.data_path, self.network, station, self.variables, station_type=self.station_type,
                         **self.kwargs)
-        print("data: ", data)
         data.interpolate(self.interpolate_dim, method=self.interpolate_method, limit=self.limit_nan_fill)
         data.transform("datetime", method=self.transform_method)
         data.make_history_window(self.interpolate_dim, self.window_history)

test/test_data_generator.py

@@ -43,13 +43,9 @@ class TestDataGenerator:
         gen.stations = ['station1', 'station2', 'station3']
         assert len(gen) == 3
 
-    def test_getitem(self, caplog, gen):
-        caplog.set_level(logging.DEBUG)
-        print(gen)
+    def test_getitem(self, gen):
         gen.kwargs = {'statistics_per_var': {'o3': 'dma8eu', 'temp': 'maximum'}}
-        print(gen)
         station = gen["DEBW107"]
-        print(station)
         assert len(station) == 2
         assert station[0].Stations.data == "DEBW107"
         assert station[0].data.shape[1:] == (8, 1, 2)

test/test_modules/test_pre_processing.py

@@ -43,7 +43,8 @@ class TestPreProcessing:
         PreProcessing()
         assert caplog.record_tuples[0] == ('root', 20, 'PreProcessing started')
         assert caplog.record_tuples[1] == ('root', 20, 'check valid stations started')
-        assert caplog.record_tuples[-2] == ('root', 20, PyTestRegex(r'run for \d+\.\d+s to check 5 station\(s\)'))
+        assert caplog.record_tuples[-2] == ('root', 20, PyTestRegex(r'run for \d+\.\d+s to check 5 station\(s\). Found '
+                                                                    r'5/5 valid stations.'))
         RunEnvironment().__del__()
 
     def test_run(self, obj_with_exp_setup):