From 45a4d7139a0304eef8820df8dbfed25a5751a24d Mon Sep 17 00:00:00 2001
From: Kate Unger <32380357+KateUnger@users.noreply.github.com>
Date: Fri, 30 Jun 2023 10:58:25 -0700
Subject: [PATCH] Fix Code Color in README

Specify code block is Python to have correct coloring in second code block in README.md
---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 8a4d852245..f3c7164095 100644
--- a/README.md
+++ b/README.md
@@ -18,7 +18,7 @@ fx.torch_to_flexflow(model, "mymodel.ff")
 
 Second, a FlexFlow program can directly import a previously saved PyTorch model and [autotune](https://www.usenix.org/conference/osdi22/presentation/unger) the parallelization performance for a given parallel machine.
 
-```
+```python
 from flexflow.pytorch.model import PyTorchModel
 
 def top_level_task():