collection failure
ImportError while importing test module '/home/runner/work/luxonis-train/luxonis-train/tests/unittests/test_core/test_archiver.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
/opt/hostedtoolcache/Python/3.10.13/x64/lib/python3.10/site-packages/optuna/integration/__init__.py:141: in _get_module
    return importlib.import_module("." + module_name, self.__name__)
/opt/hostedtoolcache/Python/3.10.13/x64/lib/python3.10/importlib/__init__.py:126: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
<frozen importlib._bootstrap>:1050: in _gcd_import
    ???
<frozen importlib._bootstrap>:1027: in _find_and_load
    ???
<frozen importlib._bootstrap>:1006: in _find_and_load_unlocked
    ???
<frozen importlib._bootstrap>:688: in _load_unlocked
    ???
<frozen importlib._bootstrap_external>:883: in exec_module
    ???
<frozen importlib._bootstrap>:241: in _call_with_frames_removed
    ???
/opt/hostedtoolcache/Python/3.10.13/x64/lib/python3.10/site-packages/optuna/integration/pytorch_lightning.py:1: in <module>
    from optuna_integration.pytorch_lightning import PyTorchLightningPruningCallback
E   ModuleNotFoundError: No module named 'optuna_integration'

During handling of the above exception, another exception occurred:
/opt/hostedtoolcache/Python/3.10.13/x64/lib/python3.10/importlib/__init__.py:126: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
tests/unittests/test_core/test_archiver.py:16: in <module>
    from luxonis_train.core import Archiver
luxonis_train/core/__init__.py:5: in <module>
    from .tuner import Tuner
luxonis_train/core/tuner.py:7: in <module>
    from optuna.integration import PyTorchLightningPruningCallback
/opt/hostedtoolcache/Python/3.10.13/x64/lib/python3.10/site-packages/optuna/integration/__init__.py:129: in __getattr__
    module = self._get_module(self._class_to_module[name])
/opt/hostedtoolcache/Python/3.10.13/x64/lib/python3.10/site-packages/optuna/integration/__init__.py:143: in _get_module
    raise ModuleNotFoundError(
E   ModuleNotFoundError: Optuna's integration modules for third-party libraries have started migrating from Optuna itself to a package called `optuna-integration`. The module you are trying to use has already been migrated to `optuna-integration`. Please install the package by running `pip install optuna-integration`.