diff --git a/ccflow/publishers/file.py b/ccflow/publishers/file.py index a0ba489b..58083ee5 100644 --- a/ccflow/publishers/file.py +++ b/ccflow/publishers/file.py @@ -4,7 +4,6 @@ import narwhals.stable.v1 as nw import pandas as pd -import yaml from cloudpathlib import AnyPath from pydantic import Field, field_validator from typing_extensions import override @@ -91,6 +90,8 @@ class YAMLPublisher(BasePublisher): @override def __call__(self) -> AnyPath: + import yaml + return GenericFilePublisher( name=self.name, name_params=self.name_params, diff --git a/ccflow/publishers/print.py b/ccflow/publishers/print.py index 8b08c957..1eff49ef 100644 --- a/ccflow/publishers/print.py +++ b/ccflow/publishers/print.py @@ -1,7 +1,6 @@ import logging from typing import Any, Generic, Literal -import yaml from pydantic import Field, field_validator from typing_extensions import override @@ -77,6 +76,8 @@ class PrintYAMLPublisher(BasePublisher): @override def __call__(self) -> Any: + import yaml + print(yaml.dump(self.data, **self.kwargs)) return self.data