No module named 'wrench.classification.self_training'

[andreaspung]

Hi, I am trying to run run_denoise.py but I am getting the following error:

Traceback (most recent call last):
  File "run_denoise.py", line 5, in <module>
    from wrench.classification import Denoise
  File "/gpfs/space/home/wrench/wrench/classification/__init__.py", line 4, in <module>
    from .self_training import LDSelfTrain, DDSelfTrain
ModuleNotFoundError: No module named 'wrench.classification.self_training'

Could you please add LDSelfTrain and DDSelfTrain classes?

[JieyuZ2]

@andreaspung hey really sry! You could just delete the line 4 in wrench/classification/init.py, these two model haven’t been ready but will be released soon!

[JieyuZ2]

fixed in latest push

[andreaspung]

Hi Jieyu, thank you! It works now 👍