Merge branch 'main' into chore/get-pr-number-from-gh-action-event-file

Author: KRRT7

.github/workflows/codeflash-optimize.yaml

@@ -34,13 +34,9 @@ jobs:
 
       - name: 📦 Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
+          uv sync
 
       - name: ⚡️Codeflash Optimization
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run codeflash --benchmark
+          uv run codeflash --benchmark

.github/workflows/end-to-end-test-benchmark-bubblesort.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
 
@@ -62,13 +61,8 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
-
+          uv sync
       - name: Run Codeflash to optimize code
         id: optimize_code_with_benchmarks
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_benchmark_sort.py
+          uv run python tests/scripts/end_to_end_test_benchmark_sort.py

.github/workflows/end-to-end-test-bubblesort-pytest-no-git.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
 
@@ -61,11 +60,8 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
-
+          uv sync
+    
       - name: Remove .git
         run: |
           if [ -d ".git" ]; then
@@ -85,5 +81,4 @@ jobs:
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_bubblesort_pytest.py
+          uv run python tests/scripts/end_to_end_test_bubblesort_pytest.py

.github/workflows/end-to-end-test-bubblesort-unittest.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
 
@@ -61,13 +60,9 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
+          uv sync
 
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_bubblesort_unittest.py
+          uv run python tests/scripts/end_to_end_test_bubblesort_unittest.py

.github/workflows/end-to-end-test-coverage.yaml

@@ -1,8 +1,7 @@
 name: Coverage E2E
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
 
@@ -59,14 +58,10 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
-          poetry add black  # my-best-repo in end_to_end_test_coverage.py is configured to use black
+          uv sync
+          uv add black  # my-best-repo in end_to_end_test_coverage.py is configured to use black
 
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_coverage.py
+          uv run python tests/scripts/end_to_end_test_coverage.py

.github/workflows/end-to-end-test-futurehouse.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
 
@@ -61,13 +60,9 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
+          uv sync
 
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_futurehouse.py
+          uv run python tests/scripts/end_to_end_test_futurehouse.py

.github/workflows/end-to-end-test-init-optim.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
   workflow_dispatch:
@@ -60,13 +59,9 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
+          uv sync
 
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_init_optimization.py
+          uv run python tests/scripts/end_to_end_test_init_optimization.py

.github/workflows/end-to-end-test-tracer-replay.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
   workflow_dispatch:
@@ -61,13 +60,9 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
+          uv sync
 
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_tracer_replay.py
+          uv run python tests/scripts/end_to_end_test_tracer_replay.py

.github/workflows/end-to-end-topological-sort-test.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
 
@@ -86,13 +85,9 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
+          uv sync
 
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_topological_sort.py
+          uv run python tests/scripts/end_to_end_test_topological_sort.py

.github/workflows/mypy.yml

@@ -21,11 +21,10 @@ jobs:
         with:
           version: "0.5.30"
 
-      - name: install poetry as a tool
-        run: uv tool install poetry
+      - name: sync uv
+        run: |
+          uv sync
 
-      - name: install dependencies
-        run: uvx poetry install --with dev
 
       - name: Run mypy on allowlist
-        run: uvx poetry run mypy --non-interactive --config-file pyproject.toml @mypy_allowlist.txt
+        run: uv run mypy --non-interactive --config-file pyproject.toml @mypy_allowlist.txt

.github/workflows/unit-tests.yaml

@@ -25,11 +25,9 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
           version: "0.5.30"
-      - name: install poetry as a tool
-        run: uv tool install poetry
 
       - name: install dependencies
-        run: uvx poetry install --with dev
+        run: uv sync
 
       - name: Unit tests
-        run: uvx poetry run pytest tests/ --benchmark-skip -m "not ci_skip"
+        run: uv run pytest tests/ --benchmark-skip -m "not ci_skip"

code_to_optimize/code_directories/simple_tracer_e2e/workload.py

@@ -21,6 +21,45 @@ def test_threadpool() -> None:
     for r in result:
         print(r)
 
+class AlexNet:
+    def __init__(self, num_classes=1000):
+        self.num_classes = num_classes
+        self.features_size = 256 * 6 * 6
+
+    def forward(self, x):
+        features = self._extract_features(x)
+        
+        output = self._classify(features)
+        return output
+
+    def _extract_features(self, x):
+        result = []
+        for i in range(len(x)):
+            pass
+
+        return result
+
+    def _classify(self, features):
+        total = sum(features)
+        return [total % self.num_classes for _ in features]
+
+class SimpleModel:
+    @staticmethod
+    def predict(data):
+        return [x * 2 for x in data]
+    
+    @classmethod
+    def create_default(cls):
+        return cls()
+
+def test_models():
+    model = AlexNet(num_classes=10)
+    input_data = [1, 2, 3, 4, 5]
+    result = model.forward(input_data)
+
+    model2 = SimpleModel.create_default()
+    prediction = model2.predict(input_data)
 
 if __name__ == "__main__":
     test_threadpool()
+    test_models()