torchfx module not installed error

[miketynes]

Sometimes (not all of the time, which is confusing) the torch fx package complains that it cannot find a submodule. Unfortunately, it doesn't tell you which one, which should be illegal (see this issue for which the PR was ignored).

See the full run trace here:

runtime.log

...
  File "/home/mike/miniconda3/envs/cascade/lib/python3.11/site-packages/ase/calculators/abc.py", line 30, in get_forces
    return self.get_property('forces', atoms)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/mike/miniconda3/envs/cascade/lib/python3.11/site-packages/ase/calculators/calculator.py", line 538, in get_property
    self.calculate(atoms, [name], system_changes)
  File "/home/mike/miniconda3/envs/cascade/lib/python3.11/site-packages/mace/calculators/mace.py", line 240, in calculate
    node_e0 = self.models[0].atomic_energies_fn(batch["node_attrs"])[
              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/mike/miniconda3/envs/cascade/lib/python3.11/site-packages/torch/fx/_symbolic_trace.py", line 800, in module_call_wrapper
    return self.call_module(mod, forward, args, kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/mike/miniconda3/envs/cascade/lib/python3.11/site-packages/torch/fx/_symbolic_trace.py", line 512, in call_module
    module_qualified_name = self.path_of_module(m)
                            ^^^^^^^^^^^^^^^^^^^^^^
  File "/home/mike/miniconda3/envs/cascade/lib/python3.11/site-packages/torch/fx/_symbolic_trace.py", line 467, in path_of_module
    raise NameError("module is not installed as a submodule")
NameError: module is not installed as a submodule