Skip to content

Commit

Permalink
linter
Browse files Browse the repository at this point in the history
  • Loading branch information
sibre28 committed Aug 19, 2024
1 parent 2bd526e commit 3505277
Showing 1 changed file with 16 additions and 16 deletions.
32 changes: 16 additions & 16 deletions src/safeds/ml/nn/_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -336,10 +336,10 @@ def fit_by_exhaustive_search(
futures.append(
executor.submit(
model.fit,
train_set,
train_set, # type: ignore[arg-type]
epoch_size,
batch_size,
learning_rate, # type: ignore[arg-type]
learning_rate,
),
) # type: ignore[arg-type]
[done, _] = wait(futures, return_when=ALL_COMPLETED)
Expand All @@ -352,10 +352,10 @@ def fit_by_exhaustive_search(
else: # train_data is TimeSeriesDataset
return self._get_best_rnn_model(
list_of_fitted_models,
train_set,
test_set,
optimization_metric, # type: ignore[arg-type]
) # type: ignore[arg-type]
train_set, # type: ignore[arg-type]
test_set, # type: ignore[arg-type]
optimization_metric,
)

def _data_split_table(self, data: TabularDataset) -> tuple[TabularDataset, TabularDataset]:
[train_split, test_split] = data.to_table().split_rows(0.75)
Expand Down Expand Up @@ -957,10 +957,10 @@ def fit_by_exhaustive_search(
futures.append(
executor.submit(
model.fit,
train_set,
train_set, # type: ignore[arg-type]
epoch_size,
batch_size,
learning_rate, # type: ignore[arg-type]
learning_rate,
),
) # type: ignore[arg-type]
[done, _] = wait(futures, return_when=ALL_COMPLETED)
Expand All @@ -975,23 +975,23 @@ def fit_by_exhaustive_search(
list_of_fitted_models,
train_set, # type: ignore[arg-type]
test_set, # type: ignore[arg-type]
optimization_metric, # type: ignore[arg-type]
optimization_metric,
positive_class,
) # type: ignore[arg-type]
)
elif isinstance(self._input_conversion, InputConversionImageToColumn):
return self._get_best_cnn_model_column(
list_of_fitted_models,
train_set,
train_set, # type: ignore[arg-type]
optimization_metric,
positive_class, # type: ignore[arg-type]
) # type: ignore[arg-type]
positive_class,
)
else: # ImageToTable
return self._get_best_cnn_model_table(
list_of_fitted_models,
train_set,
train_set, # type: ignore[arg-type]
optimization_metric,
positive_class, # type: ignore[arg-type]
) # type: ignore[arg-type]
positive_class,
)

def _data_split_table(self, data: TabularDataset) -> tuple[TabularDataset, TabularDataset]:
[train_split, test_split] = data.to_table().split_rows(0.75)
Expand Down

0 comments on commit 3505277

Please sign in to comment.