Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

fix: BasePyTorchMaterliazer -> Materializer #1969

Merged
merged 2 commits into from
Oct 30, 2023
Merged

fix: BasePyTorchMaterliazer -> Materializer #1969

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
8217105
fix: `BasePyTorchMaterliazer` -> `Materializer`
cameronraysmith Oct 20, 2023
b552cd3
Merge branch 'develop' into bugfix/basepytorchmaterializer-typo
strickvl Oct 23, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions src/zenml/integrations/neural_prophet/materializers/neural_prophet_materializer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@

from zenml.enums import ArtifactType
from zenml.integrations.pytorch.materializers.base_pytorch_materializer import (
BasePyTorchMaterliazer,
BasePyTorchMaterializer,
)

# TODO [ENG-794]: The integration consists of a simple materializer that uses the
Expand All @@ -30,7 +30,7 @@
DEFAULT_FILENAME = "entire_model.pt"


class NeuralProphetMaterializer(BasePyTorchMaterliazer):
class NeuralProphetMaterializer(BasePyTorchMaterializer):
"""Materializer to read/write NeuralProphet models."""

ASSOCIATED_TYPES: ClassVar[Tuple[Type[Any], ...]] = (NeuralProphet,)
Expand Down
2 changes: 1 addition & 1 deletion src/zenml/integrations/pytorch/materializers/base_pytorch_materializer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
DEFAULT_FILENAME = "obj.pt"


class BasePyTorchMaterliazer(BaseMaterializer):
class BasePyTorchMaterializer(BaseMaterializer):
"""Base class for PyTorch materializers."""

FILENAME: ClassVar[str] = DEFAULT_FILENAME
Expand Down
4 changes: 2 additions & 2 deletions src/zenml/integrations/pytorch/materializers/pytorch_dataloader_materializer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@

from zenml.enums import ArtifactType
from zenml.integrations.pytorch.materializers.base_pytorch_materializer import (
BasePyTorchMaterliazer,
BasePyTorchMaterializer,
)

if TYPE_CHECKING:
Expand All @@ -29,7 +29,7 @@
DEFAULT_FILENAME = "entire_dataloader.pt"


class PyTorchDataLoaderMaterializer(BasePyTorchMaterliazer):
class PyTorchDataLoaderMaterializer(BasePyTorchMaterializer):
"""Materializer to read/write PyTorch dataloaders and datasets."""

ASSOCIATED_TYPES: ClassVar[Tuple[Type[Any], ...]] = (DataLoader, Dataset)
Expand Down
4 changes: 2 additions & 2 deletions src/zenml/integrations/pytorch/materializers/pytorch_module_materializer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@

from zenml.enums import ArtifactType
from zenml.integrations.pytorch.materializers.base_pytorch_materializer import (
BasePyTorchMaterliazer,
BasePyTorchMaterializer,
)
from zenml.integrations.pytorch.utils import count_module_params
from zenml.io import fileio
Expand All @@ -34,7 +34,7 @@
CHECKPOINT_FILENAME = "checkpoint.pt"


class PyTorchModuleMaterializer(BasePyTorchMaterliazer):
class PyTorchModuleMaterializer(BasePyTorchMaterializer):
"""Materializer to read/write Pytorch models.

Inspired by the guide:
Expand Down
4 changes: 2 additions & 2 deletions src/zenml/integrations/pytorch_lightning/materializers/pytorch_lightning_materializer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,13 +19,13 @@

from zenml.enums import ArtifactType
from zenml.integrations.pytorch.materializers.base_pytorch_materializer import (
BasePyTorchMaterliazer,
BasePyTorchMaterializer,
)

CHECKPOINT_NAME = "final_checkpoint.ckpt"


class PyTorchLightningMaterializer(BasePyTorchMaterliazer):
class PyTorchLightningMaterializer(BasePyTorchMaterializer):
"""Materializer to read/write PyTorch models."""

ASSOCIATED_TYPES: ClassVar[Tuple[Type[Any], ...]] = (Module,)
Expand Down
Loading