data factory fixes

Signed-off-by: blublinsky <blublinsky@hotmail.com>

.make.versions

@@ -4,8 +4,8 @@
 ################################################################################
 
 # Data prep lab wheel version
-DPL_LIB_VERSION=0.1.3
-DPL_LIB_KFP_VERSION=0.1.4
+DPL_LIB_VERSION=0.1.4
+DPL_LIB_KFP_VERSION=0.1.5
 
 # Begin transform versions/tags
 BLOCKLIST_VERSION=0.2.0

data-processing-lib/src/data_processing/ray/transform_orchestrator.py

@@ -15,7 +15,7 @@
 from datetime import datetime
 
 import ray
-from data_processing.data_access import DataAccessFactory
+from data_processing.data_access import DataAccessFactoryBase
 from data_processing.ray import (
     DefaultTableTransformConfiguration,
     RayUtils,
@@ -34,7 +34,7 @@
 @ray.remote(num_cpus=1, scheduling_strategy="SPREAD")
 def orchestrate(
     preprocessing_params: TransformOrchestratorConfiguration,
-    data_access_factory: DataAccessFactory,
+    data_access_factory: DataAccessFactoryBase,
     transform_runtime_config: DefaultTableTransformConfiguration,
 ) -> int:
     """

data-processing-lib/src/data_processing/ray/transform_runtime.py

@@ -13,7 +13,7 @@
 import argparse
 from typing import Any
 
-from data_processing.data_access import DataAccessFactory
+from data_processing.data_access import DataAccessFactoryBase
 from data_processing.transform import AbstractTableTransform
 from data_processing.utils import CLIArgumentProvider
 from ray.actor import ActorHandle
@@ -32,7 +32,7 @@ def __init__(self, params: dict[str, Any]):
         self.params = params
 
     def get_transform_config(
-        self, data_access_factory: DataAccessFactory, statistics: ActorHandle, files: list[str]
+        self, data_access_factory: DataAccessFactoryBase, statistics: ActorHandle, files: list[str]
     ) -> dict[str, Any]:
         """
         Get the dictionary of configuration that will be provided to the transform's initializer.