map and forecast surf data

ryansurf · Jun 3, 2024 · f13c3d6 · f13c3d6 · github-actions · Jun 3, 2024
1 parent f1defec
commit f13c3d6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/cli.py b/src/cli.py
@@ -15,12 +15,15 @@
 gpt_info = [api_key, model]
 
 
-def run(lat=0, long=0):
+def run(lat=0, long=0, args=None):
     """
     Main function
     """
     # Seperates the cli args into a list
-    args = helper.seperate_args(sys.argv)
+    if args is None:
+        args = helper.seperate_args(sys.argv)
+    else:
+        args = helper.seperate_args(args)
 
     #  return coordinates, lat, long, city
     location = api.seperate_args_and_get_location(args)