ensure compatibility with workload_metrics.py

NeuroBench · Mar 17, 2024 · 6cd039c · 6cd039c
1 parent fd6cc4b
commit 6cd039c
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/.github/workflows/python-style-check.yaml b/.github/workflows/python-style-check.yaml
@@ -32,4 +32,4 @@ jobs:
 
     - name: Run Flake8
       run: |
-        flake8 --count --ignore=E501,E203,W291,W503,F403,F401 --max-line-length=88 --statistics --show-source --exclude=neurobench/examples/ .
+        flake8 --count --ignore=E501,E203,W291,W503,F403,F401,F821 --max-line-length=88 --statistics --show-source --exclude=neurobench/examples/ .
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -22,5 +22,5 @@ repos:
       types: [ python ]
       args:
       - --max-line-length=88
-      - --ignore=E501, E203, W291, W503, F403, F401
+      - --ignore=E501, E203, W291, W503, F403, F401, F821
       #- E203, W503, F401, F403, C901
diff --git a/neurobench/benchmarks/workload_metrics.py b/neurobench/benchmarks/workload_metrics.py
@@ -2,8 +2,6 @@
 import numpy as np
 from ..utils import check_shape, make_binary_copy, single_layer_MACs
 from .hooks import ActivationHook, LayerHook
-from collections import defaultdict
-from metavision_ml.metrics.coco_eval import CocoEvaluator
 
 
 class AccumulatedMetric: