From 287c7b8c8656e260dccf43c183c7717ffb3096b6 Mon Sep 17 00:00:00 2001
From: thomasw21 <24695242+thomasw21@users.noreply.github.com>
Date: Thu, 26 Aug 2021 12:24:11 +0200
Subject: [PATCH] Check cardon directory is not None

---
 megatron/global_vars.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/megatron/global_vars.py b/megatron/global_vars.py
index b5dcac4d9..1d94e8cf7 100644
--- a/megatron/global_vars.py
+++ b/megatron/global_vars.py
@@ -154,7 +154,7 @@ def _set_tensorboard_writer(args):
 
 def _set_codecarbon_tracker(args):
     global _GLOBAL_CODECARBON_TRACKER
-    if not hasattr(args, 'codecarbon_dir'):
+    if not hasattr(args, 'codecarbon_dir') or args.codecarbon_dir is None:
         return
 
     import codecarbon