From d2b72ca916a4244ba5d694fdd484e6731803837d Mon Sep 17 00:00:00 2001 From: artem Date: Mon, 14 Apr 2025 21:23:26 +0300 Subject: [PATCH] linter --- ml/cats.py | 5 +++-- ml/dogs.py | 5 +++-- ml/train.py | 2 +- 3 files changed, 7 insertions(+), 5 deletions(-) diff --git a/ml/cats.py b/ml/cats.py index 966d7d1..8ef8396 100644 --- a/ml/cats.py +++ b/ml/cats.py @@ -1,12 +1,13 @@ import json from PIL import ImageFile -ImageFile.LOAD_TRUNCATED_IMAGES = True import torch.nn as nn from torchvision.datasets import ImageFolder # type: ignore from torch.utils.data import DataLoader -from train import get_labels, load_model, get_loaders, train, show, DEVICE +from train import get_labels, load_model, get_loaders, train, show, DEVICE # type: ignore + +ImageFile.LOAD_TRUNCATED_IMAGES = True print(f"Using device: {DEVICE}") IMG_SIZE = (180, 180) diff --git a/ml/dogs.py b/ml/dogs.py index 25a81ad..adc1a99 100644 --- a/ml/dogs.py +++ b/ml/dogs.py @@ -1,12 +1,13 @@ import json from PIL import ImageFile -ImageFile.LOAD_TRUNCATED_IMAGES = True import torch.nn as nn from torchvision.datasets import ImageFolder # type: ignore from torch.utils.data import DataLoader -from train import get_labels, load_model, get_loaders, train, show, DEVICE +from train import get_labels, load_model, get_loaders, train, show, DEVICE # type: ignore + +ImageFile.LOAD_TRUNCATED_IMAGES = True print(f"Using device: {DEVICE}") IMG_SIZE = (180, 180) diff --git a/ml/train.py b/ml/train.py index c69ea19..de48614 100644 --- a/ml/train.py +++ b/ml/train.py @@ -6,7 +6,7 @@ from torchvision.datasets import ImageFolder # type: ignore from torch.utils.data import Dataset, DataLoader, random_split from torchvision import transforms # type: ignore import torchvision -from torchvision.models import ResNet50_Weights +from torchvision.models import ResNet50_Weights # type: ignore from typing import Tuple DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")