From 0cf963e8736a7dc5308d2ca1fa5d986e76accec7 Mon Sep 17 00:00:00 2001
From: "Alinson S. Xavier" <git@axavier.org>
Date: Thu, 21 Jan 2021 13:01:14 -0600
Subject: [PATCH] Fix tests for Python 3.6

---
 Makefile          | 1 +
 miplearn/types.py | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/Makefile b/Makefile
index 7d7b336..0927569 100644
--- a/Makefile
+++ b/Makefile
@@ -27,6 +27,7 @@ docs-dev:
 
 install-deps:
 	$(PIP) install -i https://pypi.gurobi.com gurobipy
+	$(PIP) install xpress
 	$(PIP) install -r requirements.txt
 
 install:
diff --git a/miplearn/types.py b/miplearn/types.py
index 4455af8..5223743 100644
--- a/miplearn/types.py
+++ b/miplearn/types.py
@@ -2,7 +2,9 @@
 #  Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved.
 #  Released under the modified BSD license. See COPYING.md for more details.
 
-from typing import TypedDict, Optional, Dict, Callable, Any
+from typing import Optional, Dict, Callable, Any
+
+from mypy_extensions import TypedDict
 
 TrainingSample = TypedDict(
     "TrainingSample",