Additional function to get home dir

src/ploomber/telemetry/telemetry.py

@@ -218,20 +218,25 @@ def parse_dag(dag):
         return None
 
 
+def get_home_dir():
+    """
+    Checks if ploomber home was set through the env variable.
+    returns the actual home_dir path.
+    """
+    return PLOOMBER_HOME_DIR if PLOOMBER_HOME_DIR else DEFAULT_HOME_DIR
+
+
 def check_dir_exist(input_location=None):
     """
     Checks if a specific directory exists, creates if not.
     In case the user didn't set a custom dir, will turn to the default home
     """
-    if PLOOMBER_HOME_DIR:
-        final_location = PLOOMBER_HOME_DIR
-    else:
-        final_location = DEFAULT_HOME_DIR
+    home_dir = get_home_dir()
 
     if input_location:
-        p = Path(final_location, input_location)
+        p = Path(home_dir, input_location)
     else:
-        p = Path(final_location)
+        p = Path(home_dir)
 
     p = p.expanduser()