diff --git a/oasis_data_manager/df_reader/config.py b/oasis_data_manager/df_reader/config.py index 952bf9e..4875a9c 100644 --- a/oasis_data_manager/df_reader/config.py +++ b/oasis_data_manager/df_reader/config.py @@ -72,7 +72,11 @@ def clean_config(config: Union[str, InputReaderConfig]) -> ResolvedReaderConfig: def get_df_reader(config, *args, **kwargs): config = clean_config(config) - cls = load_class(config["engine"]["path"], OasisReader) + if kwargs.get("df_engine", False): + cls = load_class(kwargs["df_engine"]) + else: + cls = load_class(config["engine"]["path"], OasisReader) + kwargs.pop('df_engine', None) storage = config["engine"]["options"].pop("storage", None) or LocalStorage("/") return cls(