model.export() fails with ModuleNotFoundError: No module named 'wandb'

[robmarkcole]

Search before asking

• I have searched the RF-DETR issues and found no similar bug report.

Bug

Export shouldn't depend on the logger.
Trace:

ModuleNotFoundError                       Traceback (most recent call last)
Cell In[16], line 1
----> 1 model.export()

File /home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/rfdetr/detr.py:39, in RFDETR.export(self, **kwargs)
     38 def export(self, **kwargs):
---> 39     self.model.export(**kwargs)

File /home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/rfdetr/main.py:446, in Model.export(self, output_dir, infer_dir, simplify, backbone_only, opset_version, verbose, force, shape, batch_size, **kwargs)
    444 """Export the trained model to ONNX format"""
    445 print(f"Exporting model to ONNX format")
--> 446 from rfdetr.deploy.export import export_onnx, onnx_simplify, make_infer_image
    449 device = self.device
    450 model = deepcopy(self.model.to("cpu"))

File /home/zeus/miniconda3/envs/cloudspace/lib/python3.10/site-packages/rfdetr/deploy/export.py:32
     30 import re
     31 import sys
---> 32 import wandb

ModuleNotFoundError: No module named 'wandb'

Environment

NA

Minimal Reproducible Example

model.export

Additional

No response

Are you willing to submit a PR?

• Yes, I'd like to help by submitting a PR!

[nok]

Hi @robmarkcole, just a quick note - this bug has already been fixed in these commits on the develop branch: 611ab26 , 429cff2

As a workaround, you can install the develop branch directly using:

pip install git+ssh://git@github.com/roboflow/rf-detr.git@develop