🐛[BUG]: Missing dependency

[ThomasMGeo]

Version

'0.2.0a0'

On which installation method(s) does this occur?

Source

Describe the issue

Working through the notebook / .py file, when I get to this block:

`# Option 1: Use config file and CLI (use this outside a notebook)

with open('./01_config.json', 'w') as f:

json.dump(config, f)

! python3 -m earth2mip.inference_ensemble 01_config.json

Option 2: Feed in JSON string directly into main function

config_str = json.dumps(config)
inference_ensemble.main(config_str)`

I get this traceback error:

ModuleNotFoundError: No module named 'apex'

I am assuming it's this package, but might be worth adding to pyproject.toml.

Environment details