profiling

KoslickiLab · Feb 5, 2024 · 5ac0214 · 5ac0214
1 parent 137dc9b
commit 5ac0214
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/.github/workflows/runTest.yml b/.github/workflows/runTest.yml
@@ -29,4 +29,6 @@ jobs:
         CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
     - name: Badge Action
       uses: emibcn/[email protected]
+    - name: profiling
+      run: cat yacht.pclprof  
 
diff --git a/env/yacht_env.yml b/env/yacht_env.yml
@@ -23,3 +23,4 @@ dependencies:
   - sourmash_plugin_branchwater
   - pip:
     - openpyxl
+    - line-profiler-pycharm
diff --git a/yacht/make_training_data_from_sketches.py b/yacht/make_training_data_from_sketches.py
@@ -7,6 +7,7 @@
 import json
 import shutil
 from . import utils
+from line_profiler_pycharm import profile
 
 # Configure Loguru logger
 logger.remove()
@@ -26,6 +27,7 @@ def add_arguments(parser):
     parser.add_argument('--outdir', type=str, help='Path to output directory.', required=False, default=os.getcwd())
     parser.add_argument('--force', action='store_true', help='Overwrite the output directory if it exists.')
 
+@profile
 def main(args):
     # get the arguments
     ref_file = str(Path(args.ref_file).absolute())