From 454faa8e13529b4688fddfd72ae4d82c64b82e99 Mon Sep 17 00:00:00 2001
From: Mihir Patel <mihir.v.patel7@gmail.com>
Date: Thu, 7 Dec 2023 17:18:01 -0500
Subject: [PATCH] Disable mosaicml logger in foundry CI/CD (#788)

* logs

* change to env var

* lint

* lint
---
 .github/mcp/mcp_pytest.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/.github/mcp/mcp_pytest.py b/.github/mcp/mcp_pytest.py
index b6d74880c8..7571200d8d 100644
--- a/.github/mcp/mcp_pytest.py
+++ b/.github/mcp/mcp_pytest.py
@@ -114,6 +114,16 @@
         integrations=[git_integration],
         command=command,
         scheduling={'max_duration': args.timeout / 60 / 60},
+        env_variables=[
+            {
+                'key': 'MOSAICML_PLATFORM',
+                'value': 'False',
+            },
+            {
+                'key': 'PYTHONUNBUFFERED',
+                'value': '1',
+            },
+        ],
     )
 
     # Create run