Automated autopep8 fixes

aimclub · Jun 27, 2024 · de02ad3 · de02ad3
1 parent b3f9cd8
commit de02ad3
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/fedot_ind/core/models/nn/network_impl/tst.py b/fedot_ind/core/models/nn/network_impl/tst.py
@@ -303,9 +303,9 @@ def _init_model(self, ts):
 
         'quantile_rf_model': PipelineBuilder().add_node('quantile_extractor').add_node('rf'),
 
-        'composed_model': PipelineBuilder().add_node('tst_model', params={'epochs': 50, 'batch_size': 32})\
-                                           .add_node('quantile_extractor', branch_idx=1)\
-                                           .add_node('rf', branch_idx=1)\
+        'composed_model': PipelineBuilder().add_node('tst_model', params={'epochs': 50, 'batch_size': 32})
+                                           .add_node('quantile_extractor', branch_idx=1)
+                                           .add_node('rf', branch_idx=1)
                                            .join_branches('logit')}
 
     train_data, test_data = DataLoader(dataset_list[0]).load_data()

diff --git a/fedot_ind/core/models/nn/network_modules/layers/conv_layers.py b/fedot_ind/core/models/nn/network_modules/layers/conv_layers.py
@@ -20,9 +20,9 @@ class Conv2dSame(Module):
 
     def __init__(
             self, ni, nf, ks=(
-                    3, 3), stride=(
-                    1, 1), dilation=(
-                    1, 1), **kwargs):
+                3, 3), stride=(
+                1, 1), dilation=(
+                1, 1), **kwargs):
         if isinstance(ks, Integral):
             ks = (ks, ks)
         if isinstance(stride, Integral):
@@ -57,7 +57,7 @@ def Conv2d(
         **kwargs):
     """conv1d layer with padding='same', 'valid', or any integer (defaults to 'same')"""
     assert not (
-            kernel_size and ks), 'use kernel_size or ks but not both simultaneously'
+        kernel_size and ks), 'use kernel_size or ks but not both simultaneously'
     assert kernel_size is not None or ks is not None, 'you need to pass a ks'
     kernel_size = kernel_size or ks
     if padding == 'same':

diff --git a/tests/unit/core/models/model_impl/test_omni_scale.py b/tests/unit/core/models/model_impl/test_omni_scale.py
@@ -31,5 +31,5 @@ def test_omniscale_model(ts_input_data):
                                            ).build()
 
         model.fit(train)
-        predict = model.predict(test)
+        model.predict(test)
         assert model is not None