Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] authored and Borda committed Nov 3, 2021
1 parent 4747f09 commit c775078
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 4 deletions.
3 changes: 1 addition & 2 deletions flash/core/Regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,6 @@


class RegressionMixin:

@staticmethod
def _build(
loss_fn: Optional[Callable] = None,
Expand Down Expand Up @@ -57,4 +56,4 @@ def __init__(
metrics=metrics,
serializer=serializer,
**kwargs,
)
)
1 change: 0 additions & 1 deletion flash/core/classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down
2 changes: 1 addition & 1 deletion flash/tabular/regression/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit c775078

Please sign in to comment.