Skip to content

Commit

Permalink
fix lint error
Browse files Browse the repository at this point in the history
  • Loading branch information
dpower4 committed Dec 19, 2024
1 parent bf780d0 commit 48e67f8
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion superbench/benchmarks/model_benchmarks/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
"""A module containing all the e2e model related benchmarks."""

import sys
from typing import Optional

from superbench.benchmarks.model_benchmarks.model_base import ModelBenchmark
from superbench.benchmarks.model_benchmarks.pytorch_bert import PytorchBERT
Expand All @@ -14,7 +15,7 @@
from superbench.benchmarks.model_benchmarks.pytorch_llama import PytorchLlama

# Check for Python version > 3.7 and conditionally import PytorchMixtral
PytorchMixtral = None
PytorchMixtral: Optional[type] = None
if sys.version_info > (3, 7):
from superbench.benchmarks.model_benchmarks.pytorch_mixtral import PytorchMixtral

Expand Down

0 comments on commit 48e67f8

Please sign in to comment.