torchani 0.2.1__tar.gz → 0.4__tar.gz

opt/hostedtoolcache/Python/3.7.0/x64/lib/python3.7/site-packages/torchani/__init__.py

@@ -27,11 +27,14 @@ at :attr:`torchani.ignite`, and more at :attr:`torchani.utils`.
 from .utils import EnergyShifter
 from .nn import ANIModel, Ensemble
 from .aev import AEVComputer
-from . import ignite
 from . import utils
 from . import neurochem
-from . import data
+from . import models
 from pkg_resources import get_distribution, DistributionNotFound
+import sys
+if sys.version_info[0] > 2:
+    from . import ignite
+    from . import data
 
 try:
     __version__ = get_distribution(__name__).version
@@ -40,7 +43,7 @@ except DistributionNotFound:
     pass
 
 __all__ = ['AEVComputer', 'EnergyShifter', 'ANIModel', 'Ensemble',
-           'ignite', 'utils', 'neurochem', 'data']
+           'ignite', 'utils', 'neurochem', 'data', 'models']
 
 try:
     from . import ase  # noqa: F401

opt/hostedtoolcache/Python/3.7.0/x64/lib/python3.7/site-packages/torchani/_six.py

@@ -0,0 +1,5 @@
+import math
+
+
+if not hasattr(math, 'inf'):
+    math.inf = float('inf')