Make DLLogger less mandatory

openfold/utils/logger.py

@@ -11,17 +11,21 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
 import os
 import operator
 import time
 
-import dllogger as logger
 import numpy as np
 import torch.cuda.profiler as profiler
-from dllogger import JSONStreamBackend, StdOutBackend, Verbosity
 from pytorch_lightning import Callback
 
+# We make this optional for the Colab's sake
+try:
+    import dllogger as logger
+    from dllogger import JSONStreamBackend, StdOutBackend, Verbosity
+except:
+    pass
+
 
 def is_main_process():
     return int(os.getenv("LOCAL_RANK", "0")) == 0