From c2896f3b5b29295ae25c9ad3323e154757edf84e Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Wed, 4 Dec 2024 18:45:54 +0100
Subject: [PATCH] Added basic profiling

---
 .github/workflows/profiling.yaml | 11 +++--------
 1 file changed, 3 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/profiling.yaml b/.github/workflows/profiling.yaml
index 15ad96303..a6214ebff 100644
--- a/.github/workflows/profiling.yaml
+++ b/.github/workflows/profiling.yaml
@@ -29,12 +29,7 @@ jobs:
     - name: Install dependencies
       run: |
         poetry install --no-interaction --all-extras
-
-    # Install necessary dependencies
-    - name: Install dependencies
-      run: |
-        python -m pip install --upgrade pip
-        pip install scalene requests
+        poetry add scalene requests
 
 
     # Set environment variables for SHAs
@@ -55,7 +50,7 @@ jobs:
         # Ensure the script is executable
         chmod +x cognee/api/v1/cognify/code_graph_pipeline.py
         # Run Scalene
-        scalene --json --outfile base_results.json cognee/api/v1/cognify/code_graph_pipeline.py
+        poetry run scalene --json --outfile base_results.json cognee/api/v1/cognify/code_graph_pipeline.py
 
     # Run profiler on head branch
     - name: Run profiler on head branch
@@ -69,7 +64,7 @@ jobs:
         # Ensure the script is executable
         chmod +x cognee/api/v1/cognify/code_graph_pipeline.py
         # Run Scalene
-        scalene --json --outfile head_results.json cognee/api/v1/cognify/code_graph_pipeline.py
+        poetry run scalene --json --outfile head_results.json cognee/api/v1/cognify/code_graph_pipeline.py
 
     # Compare profiling results
     - name: Compare profiling results