Skip to content

Commit

Permalink
fix(joblib): update pipe function to use BATCHER instead of PIPE in H…
Browse files Browse the repository at this point in the history
…yFI class
  • Loading branch information
entelecheia committed Jun 24, 2023
1 parent bc6a96b commit cd3c94c
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/hyfi/main/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
from hyfi.composer import Composer, DictKeyType, SpecialKeys
from hyfi.composer.extended import XC
from hyfi.dotenv import DotEnvConfig
from hyfi.joblib import JobLibConfig
from hyfi.joblib import JobLibConfig, BATCHER
from hyfi.pipe import PIPE, PipeConfig
from hyfi.project import ProjectConfig
from hyfi.utils.datasets import DatasetLikeType, Datasets, DatasetType
Expand Down Expand Up @@ -374,8 +374,8 @@ def setLogger(level=None, force=True, **kwargs):
# Batcher related functions
###############################
@staticmethod
def pipe(data: Any, pipe_config: Union[Dict, PipeConfig]):
return PIPE.pipe(data, pipe_config)
def pipe(obj: Any, pipe_config: Union[Dict, PipeConfig]):
return PIPE.pipe(obj, pipe_config)

@staticmethod
def apply(
Expand All @@ -387,7 +387,7 @@ def apply(
num_workers: Optional[int] = None,
**kwargs,
):
return PIPE.apply(
return BATCHER.apply(
func,
series,
description=description,
Expand Down

0 comments on commit cd3c94c

Please sign in to comment.