fix imports

This commit is contained in:
Yinon Polak
2023-04-03 16:03:15 +03:00
parent bd3b70293f
commit c137666230
6 changed files with 14 additions and 11 deletions

View File

@@ -1,5 +1,5 @@
import logging
from typing import Tuple, List
from typing import List
import torch
import torch.nn as nn

View File

@@ -12,6 +12,7 @@ from torch.utils.data import DataLoader, TensorDataset
from freqtrade.freqai.torch.PyTorchDataConvertor import PyTorchDataConvertor
from freqtrade.freqai.torch.PyTorchTrainerInterface import PyTorchTrainerInterface
logger = logging.getLogger(__name__)

View File

@@ -1,12 +1,11 @@
from abc import ABC, abstractmethod
from typing import Any, Dict, List, Optional, Tuple
from pathlib import Path
from typing import Dict, List
import pandas as pd
import torch
import torch.nn as nn
from pathlib import Path
class PyTorchTrainerInterface(ABC):
@@ -51,4 +50,4 @@ class PyTorchTrainerInterface(ABC):
get_init_model method.
:checkpoint checkpoint: dict containing the model & optimizer state dicts,
model_meta_data, etc..
"""
"""