Skip to content

Commit a06d06e

Browse files
committed
move trainerABC to separate file
1 parent e9d651c commit a06d06e

File tree

3 files changed

+11
-9
lines changed

3 files changed

+11
-9
lines changed

transfer_nlp/plugins/trainer_abc.py

+8
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
from abc import ABC, abstractmethod
2+
3+
4+
class TrainerABC(ABC):
5+
6+
@abstractmethod
7+
def train(self):
8+
pass

transfer_nlp/plugins/trainers.py

+2-8
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
import inspect
1212
import logging
1313
import re
14-
from abc import ABC, abstractmethod
14+
from abc import abstractmethod
1515
from collections import defaultdict
1616
from typing import Dict, List, Any, Union, Tuple
1717

@@ -30,6 +30,7 @@
3030
from transfer_nlp.loaders.loaders import DatasetSplits
3131
from transfer_nlp.plugins.config import register_plugin
3232
from transfer_nlp.plugins.regularizers import RegularizerABC
33+
from transfer_nlp.plugins.trainer_abc import TrainerABC
3334

3435
logger = logging.getLogger(__name__)
3536

@@ -94,13 +95,6 @@ def compute(self):
9495
return self.source_metric.compute()
9596

9697

97-
class TrainerABC(ABC):
98-
99-
@abstractmethod
100-
def train(self):
101-
pass
102-
103-
10498
@register_plugin
10599
class BaseIgniteTrainer(TrainerABC):
106100

transfer_nlp/runner/experiment_runner.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
from transfer_nlp.plugins.config import ExperimentConfig
1111
from transfer_nlp.plugins.reporters import ReporterABC
12-
from transfer_nlp.plugins.trainers import TrainerABC
12+
from transfer_nlp.plugins.trainer_abc import TrainerABC
1313

1414
ConfigEnv = Dict[str, Any]
1515

0 commit comments

Comments
 (0)