feat(pm4py): timestamp configuration changes

process-intelligence-solutions · Nov 28, 2023 · 4feb5c2 · 4feb5c2
1 parent b13ea64
commit 4feb5c2
Show file tree

Hide file tree

Showing 3 changed files with 22,390 additions and 22,367 deletions.
diff --git a/pm4py/util/constants.py b/pm4py/util/constants.py
@@ -8,13 +8,31 @@ def get_param_from_env(name, default):
         return str(os.environ[name])
     return default
 
+
+TEST_CUDF_DATAFRAMES_ENVIRONMENT = get_param_from_env("PM4PY_TEST_CUDF_DATAFRAMES_ENVIRONMENT", False)
+
+
 def get_default_timestamp_format():
+    if importlib.util.find_spec("cudf") or TEST_CUDF_DATAFRAMES_ENVIRONMENT:
+        return "%Y-%m-%d %H:%M:%S"
+        pass
+
     return None
 
+
 def get_default_xes_timestamp_format():
+    if importlib.util.find_spec("cudf") or TEST_CUDF_DATAFRAMES_ENVIRONMENT:
+        return "%Y-%m-%dT%H:%M:%S"
+        pass
+
     return "ISO8601"
 
+
 def get_default_is_aware_enabled():
+    if importlib.util.find_spec("cudf") or TEST_CUDF_DATAFRAMES_ENVIRONMENT:
+        return False
+        pass
+
     return True
 
 

diff --git a/pm4py/util/pandas_utils.py b/pm4py/util/pandas_utils.py
@@ -286,6 +286,11 @@ def dataframe_column_string_to_datetime(*args, **kwargs):
 
 
 def read_csv(*args, **kwargs):
+    if kwargs:
+        if importlib.util.find_spec("cudf") or constants.TEST_CUDF_DATAFRAMES_ENVIRONMENT:
+            if "encoding" in kwargs:
+                del kwargs["encoding"]
+
     return DATAFRAME.read_csv(*args, **kwargs)