[HOPSWORKS-2122] Backend not handling null TimeTravelFormat sent by c…

…lient (#133)
logicalclocks · Nov 6, 2020 · 23777d1 · 23777d1
1 parent 2665497
commit 23777d1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/hsfs/feature_group.py b/python/hsfs/feature_group.py
@@ -30,7 +30,6 @@ class FeatureGroup:
     CACHED_FEATURE_GROUP = "CACHED_FEATURE_GROUP"
     ON_DEMAND_FEATURE_GROUP = "ON_DEMAND_FEATURE_GROUP"
     ENTITY_TYPE = "featuregroups"
-    HUDI = "HUDI"
 
     def __init__(
         self,
@@ -52,7 +51,7 @@ def __init__(
         feat_hist_enabled=None,
         statistic_columns=None,
         online_enabled=False,
-        time_travel_format=HUDI,
+        time_travel_format=None,
         hudi_enabled=False,
         statistics_config=None,
     ):
@@ -71,7 +70,9 @@ def __init__(
         self._location = location
         self._jobs = jobs
         self._online_enabled = online_enabled
-        self._time_travel_format = time_travel_format
+        self._time_travel_format = (
+            time_travel_format.upper() if time_travel_format is not None else None
+        )
         self._hudi_enabled = hudi_enabled
 
         if id is not None: