Ayush Chaurasia commited on
Commit
8f6e447
1 Parent(s): a57f23d

Fix Indentation in test.py (#2614)

Browse files

* Fix Indentation in test.py

* CI fix

* Comply with PEP8: 80 characters per line

Files changed (1) hide show
  1. test.py +1 -1
test.py CHANGED
@@ -156,7 +156,7 @@ def test(data,
156
  "domain": "pixel"} for *xyxy, conf, cls in pred.tolist()]
157
  boxes = {"predictions": {"box_data": box_data, "class_labels": names}} # inference-space
158
  wandb_images.append(wandb_logger.wandb.Image(img[si], boxes=boxes, caption=path.name))
159
- wandb_logger.log_training_progress(predn, path, names) # logs dsviz tables
160
 
161
  # Append to pycocotools JSON dictionary
162
  if save_json:
 
156
  "domain": "pixel"} for *xyxy, conf, cls in pred.tolist()]
157
  boxes = {"predictions": {"box_data": box_data, "class_labels": names}} # inference-space
158
  wandb_images.append(wandb_logger.wandb.Image(img[si], boxes=boxes, caption=path.name))
159
+ wandb_logger.log_training_progress(predn, path, names) if wandb_logger and wandb_logger.wandb_run else None
160
 
161
  # Append to pycocotools JSON dictionary
162
  if save_json: