From 6fa074168e2bbd6e56a6bb5212d606c81fb1bb7c Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin@meta.com>
Date: Thu, 29 Aug 2024 16:14:45 -0700
Subject: [PATCH] update paths

---
 llama_toolchain/cli/api/start.py   | 4 ++--
 llama_toolchain/cli/stack/start.py | 4 ++--
 llama_toolchain/core/package.py    | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/llama_toolchain/cli/api/start.py b/llama_toolchain/cli/api/start.py
index 539679561..a162d8dcb 100644
--- a/llama_toolchain/cli/api/start.py
+++ b/llama_toolchain/cli/api/start.py
@@ -62,13 +62,13 @@ class ApiStart(Subcommand):
         if config.docker_image:
             script = pkg_resources.resource_filename(
                 "llama_toolchain",
-                "distribution/start_container.sh",
+                "core/start_container.sh",
             )
             run_args = [script, config.docker_image]
         else:
             script = pkg_resources.resource_filename(
                 "llama_toolchain",
-                "distribution/start_conda_env.sh",
+                "core/start_conda_env.sh",
             )
             run_args = [
                 script,
diff --git a/llama_toolchain/cli/stack/start.py b/llama_toolchain/cli/stack/start.py
index 6b8981128..433d5b8dc 100644
--- a/llama_toolchain/cli/stack/start.py
+++ b/llama_toolchain/cli/stack/start.py
@@ -70,13 +70,13 @@ class StackStart(Subcommand):
         if config.docker_image:
             script = pkg_resources.resource_filename(
                 "llama_toolchain",
-                "distribution/start_container.sh",
+                "core/start_container.sh",
             )
             run_args = [script, config.docker_image]
         else:
             script = pkg_resources.resource_filename(
                 "llama_toolchain",
-                "distribution/start_conda_env.sh",
+                "core/start_conda_env.sh",
             )
             run_args = [
                 script,
diff --git a/llama_toolchain/core/package.py b/llama_toolchain/core/package.py
index 9cb58569d..7ef68bcc9 100644
--- a/llama_toolchain/core/package.py
+++ b/llama_toolchain/core/package.py
@@ -147,7 +147,7 @@ def build_package(
 
     if build_type == BuildType.container:
         script = pkg_resources.resource_filename(
-            "llama_toolchain", "distribution/build_container.sh"
+            "llama_toolchain", "core/build_container.sh"
         )
         args = [
             script,
@@ -158,7 +158,7 @@ def build_package(
         ]
     else:
         script = pkg_resources.resource_filename(
-            "llama_toolchain", "distribution/build_conda_env.sh"
+            "llama_toolchain", "core/build_conda_env.sh"
         )
         args = [
             script,