24912462866

Committed 24 Apr 2026 09:21PM UTC coverage: 76.362% (-3.1%) from 79.464%

Build # 24912462866

Build Type

push

github

Committed by

web-flow

Commit Message

Merge pull request #73 from rafaelpadilla/eduardo/refactor_data_operations

Refactor of data operations, trainers and models.

Coverage Stats

244 of 339 branches covered (71.98%)

Branch coverage included in aggregate %.

1317 of 1706 new or added lines in 50 files covered. (77.2%)

28 existing lines in 5 files now uncovered.

2124 of 2762 relevant lines covered (76.9%)

0.77 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

93.33

/toolkit/ThreeWToolkit/core/base_preprocessing.py

from pydantic import BaseModel
from abc import ABC, abstractmethod
from .base_dataset import BaseDataset
from .base_instantiable import Instantiable
from .dataset_outputs import DatasetOutputs


class BasePreprocessingConfig(BaseModel, Instantiable):
    """Base configuration for preprocessing steps."""

    _target: type["BasePreprocessing"]


class BasePreprocessing(ABC):
    """Base class for preprocessing steps."""

    def __init__(self, config: BasePreprocessingConfig):
        self.config = config

    @abstractmethod
    def transform(self, data: DatasetOutputs) -> DatasetOutputs:
        """Transform the data using the fitted preprocessing step.

        Args:
            data: Input dataset outputs to transform.

        Returns:
            Transformed dataset outputs.
        """
        raise NotImplementedError("Subclasses must implement the transform method.")

    def fit(self, data: BaseDataset) -> None:
        """Fit the preprocessing step to the dataset if needed.

        Args:
            data: Dataset to fit the step on.
        """
        pass  # do nothing by default.

1	from pydantic import BaseModel	1✔
2	from abc import ABC, abstractmethod	1✔
3	from .base_dataset import BaseDataset	1✔
4	from .base_instantiable import Instantiable	1✔
5	from .dataset_outputs import DatasetOutputs	1✔
6
7
8	class BasePreprocessingConfig(BaseModel, Instantiable):	1✔
9	"""Base configuration for preprocessing steps."""
10
11	_target: type["BasePreprocessing"]	1✔
12
13
14	class BasePreprocessing(ABC):	1✔
15	"""Base class for preprocessing steps."""
16
17	def __init__(self, config: BasePreprocessingConfig):	1✔
18	self.config = config	1✔
19
20	@abstractmethod	1✔
21	def transform(self, data: DatasetOutputs) -> DatasetOutputs:	1✔
22	"""Transform the data using the fitted preprocessing step.
23
24	Args:
25	data: Input dataset outputs to transform.
26
27	Returns:
28	Transformed dataset outputs.
29	"""
NEW 30	raise NotImplementedError("Subclasses must implement the transform method.")	×
31
32	def fit(self, data: BaseDataset) -> None:	1✔
33	"""Fit the preprocessing step to the dataset if needed.
34
35	Args:
36	data: Dataset to fit the step on.
37	"""
38	pass # do nothing by default.	1✔

rafaelpadilla / 3W / 24912462866

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous