Update engine.py

sdtblck · web-flow · commit faae906af270 · 2021-05-19T14:52:32.000+02:00
diff --git a/deepspeed/runtime/pipe/engine.py b/deepspeed/runtime/pipe/engine.py
@@ -199,7 +199,7 @@ def __init__(self, *super_args, **super_kwargs):
                 p2p.send(self.loss, self.next_stage)
 
     def set_has_attention_mask(self, value):
-        assert isinstance(value, boolean)
+        assert isinstance(value, bool)
         self.has_attention_mask = value
         
     def _build_data_iter(self, dataset):
