Fixed TensorImage being treated as unsupported despite having its own code branch
@@ -164,7 +164,7 @@ class WandBLogger(BaseLogger, SupervisedPlugin):
if not isinstance(
value,
- (Image, Tensor, Figure, float, int, self.wandb.viz.CustomChart),
+ (Image, TensorImage, Tensor, Figure, float, int, self.wandb.viz.CustomChart),
):
# Unsupported type
return
