From 097fd22f4ccbb7c02c9a1166285280ce22f6083d Mon Sep 17 00:00:00 2001
From: Jirka Borovec <jirka.borovec@seznam.cz>
Date: Mon, 19 Oct 2020 20:10:20 +0200
Subject: [PATCH] imports

---
 pl_bolts/callbacks/self_supervised.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pl_bolts/callbacks/self_supervised.py b/pl_bolts/callbacks/self_supervised.py
index b6133620a0..5c133f10b0 100644
--- a/pl_bolts/callbacks/self_supervised.py
+++ b/pl_bolts/callbacks/self_supervised.py
@@ -103,7 +103,7 @@ def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, data
         pl_module.logger.log_metrics(metrics, step=trainer.global_step)
 
 
-class BYOLMAWeightUpdate(pl.Callback):
+class BYOLMAWeightUpdate(Callback):
     """
     Weight update rule from BYOL.