final updates ahead of initial publication

databricks-industry-solutions · Aug 8, 2024 · 94a2f65 · 94a2f65
1 parent e42ba34
commit 94a2f65
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/04a_timegpt_load_inference.py b/04a_timegpt_load_inference.py
@@ -36,6 +36,22 @@
 
 # COMMAND ----------
 
+# MAGIC %md
+# MAGIC ##Set the logging level
+
+# COMMAND ----------
+
+import logging
+
+logger = spark._jvm.org.apache.log4j
+
+# Setting the logging level to ERROR for the "py4j.java_gateway" logger
+# This reduces the verbosity of the logs by only showing error messages
+logging.getLogger("py4j.java_gateway").setLevel(logging.ERROR)
+logging.getLogger("py4j.clientserver").setLevel(logging.ERROR)
+
+# COMMAND ----------
+
 # MAGIC %md
 # MAGIC ## Record the TimeGen-1 Endpoint Target URI
 

diff --git a/04b_timegpt_fine_tune.py b/04b_timegpt_fine_tune.py
@@ -35,6 +35,22 @@
 
 # COMMAND ----------
 
+# MAGIC %md
+# MAGIC ##Set the logging level
+
+# COMMAND ----------
+
+import logging
+
+logger = spark._jvm.org.apache.log4j
+
+# Setting the logging level to ERROR for the "py4j.java_gateway" logger
+# This reduces the verbosity of the logs by only showing error messages
+logging.getLogger("py4j.java_gateway").setLevel(logging.ERROR)
+logging.getLogger("py4j.clientserver").setLevel(logging.ERROR)
+
+# COMMAND ----------
+
 # MAGIC %md
 # MAGIC ## Record the TimeGen-1 Endpoint Target URI