Update fake classes for type hints

metatensor · Jul 5, 2024 · 94e0973 · 94e0973
1 parent c660e07
commit 94e0973
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/python/metatensor-torch/metatensor/torch/atomistic/openmm_force.py b/python/metatensor-torch/metatensor/torch/atomistic/openmm_force.py
@@ -20,14 +20,16 @@
  HAS_OPENMM = True
 except ImportError:
 
- class MLPotential:
- pass
+ class openmm:
+ class System:
+ pass
 
- class MLPotentialImpl:
- pass
+  class Force:
+  pass
 
- class MLPotentialImplFactory:
- pass
+ class app:
+ class Topology:
+ pass
 
  HAS_OPENMM = False