Address PR Review Comments

Author: Jonathan Makunga

src/sagemaker/image_uri_config/tasks.json

@@ -64,4 +64,4 @@
 			]
 		}
     }
-}
+}

src/sagemaker/serve/utils/task.py

@@ -31,9 +31,8 @@ def retrieve_local_schemas(task: str) -> Tuple[Any, Any]:
     Raises:
         ValueError: If no tasks config found or the task does not exist in the local config.
     """
-    # task_io_config_path = os.path.join(os.path.dirname(__file__), "image_uri_config", "tasks.json")
-    c = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))
-    task_io_config_path = os.path.join(c, "image_uri_config", "tasks.json")
+    config_dir = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))
+    task_io_config_path = os.path.join(config_dir, "image_uri_config", "tasks.json")
     try:
         with open(task_io_config_path) as f:
             task_io_config = json.load(f)