From 9e390e801b945a1dfd60ebb751142e117ea4544e Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 27 Oct 2021 00:03:37 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- flash/core/Regression.py | 3 +-- flash/core/classification.py | 1 - flash/tabular/regression/model.py | 2 +- 3 files changed, 2 insertions(+), 4 deletions(-) diff --git a/flash/core/Regression.py b/flash/core/Regression.py index 7d103e75a3..8cafcb5472 100644 --- a/flash/core/Regression.py +++ b/flash/core/Regression.py @@ -22,7 +22,6 @@ class RegressionMixin: - @staticmethod def _build( loss_fn: Optional[Callable] = None, @@ -57,4 +56,4 @@ def __init__( metrics=metrics, serializer=serializer, **kwargs, - ) \ No newline at end of file + ) diff --git a/flash/core/classification.py b/flash/core/classification.py index 0d0c647fd6..884dffbdfb 100644 --- a/flash/core/classification.py +++ b/flash/core/classification.py @@ -39,7 +39,6 @@ def binary_cross_entropy_with_logits(x: torch.Tensor, y: torch.Tensor) -> torch. class ClassificationMixin: - @staticmethod def _build( num_classes: Optional[int] = None, diff --git a/flash/tabular/regression/model.py b/flash/tabular/regression/model.py index 66ba9e1dd3..95f1ecf296 100644 --- a/flash/tabular/regression/model.py +++ b/flash/tabular/regression/model.py @@ -16,8 +16,8 @@ import torch from torch.nn import functional as F -from flash.core.Regression import RegressionTask from flash.core.data.data_source import DefaultDataKeys +from flash.core.Regression import RegressionTask from flash.core.utilities.imports import _TABULAR_AVAILABLE from flash.core.utilities.types import LR_SCHEDULER_TYPE, METRICS_TYPE, OPTIMIZER_TYPE, SERIALIZER_TYPE