Skip to content

Commit

Permalink
(ME) small adjustment to the last commit
Browse files Browse the repository at this point in the history
  • Loading branch information
amkrajewski committed Feb 17, 2024
1 parent fcd09ca commit fa150f6
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions pysipfenn/core/modelExporters.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,8 @@ def __init__(self, calculator: Calculator):
from onnxconverter_common import float16
from onnxsim import simplify
except ModuleNotFoundError as e:
raise Exception(str(e) + '\n\nNote: Export Dependencies are not installed by default. If you need them, you have to install '
'pySIPFENN in `dev` mode like: `pip install "pysipfenn[dev]"`, or like `pip install -e ".[dev]"` (see pysipfenn.org)')
print('\n\nNote: Export Dependencies are not installed by default. If you need them, you have to install pySIPFENN in '
'`dev` mode like: `pip install "pysipfenn[dev]"`, or like `pip install -e ".[dev]"` (see pysipfenn.org)')

self.simplifiedDict = {model: False for model in calculator.loadedModels.keys()}
self.fp16Dict = {model: False for model in calculator.loadedModels.keys()}
Expand Down Expand Up @@ -242,9 +242,8 @@ def __init__(self, calculator: Calculator):
try:
import coremltools as ct
except ModuleNotFoundError as e:
raise Exception(str(e) + '\n\nNote: Export Dependencies are not installed by default. If you need them, you have to install '
'pySIPFENN in `dev` mode like: `pip install "pysipfenn[dev]"`, or like `pip install -e ".[dev]"` (see pysipfenn.org)')

print('\n\nNote: Export Dependencies are not installed by default. If you need them, you have to install pySIPFENN in '
'`dev` mode like: `pip install "pysipfenn[dev]"`, or like `pip install -e ".[dev]"` (see pysipfenn.org)')

self.calculator = calculator
assert len(self.calculator.loadedModels)>0, 'No models loaded in calculator. Nothing to export.'
Expand Down

0 comments on commit fa150f6

Please sign in to comment.