From 826137bd3b4e2aafae9bb6ac3d88bbf774a438a4 Mon Sep 17 00:00:00 2001
From: yangdsh <dongshen@andrew.cmu.edu>
Date: Tue, 22 Oct 2019 23:09:56 +0000
Subject: [PATCH] add simulation to white-list

---
 script/validators/source_validator.py | 2 ++
 server/analysis/simulation.py         | 2 --
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/script/validators/source_validator.py b/script/validators/source_validator.py
index 15b99c0..17d8a82 100644
--- a/script/validators/source_validator.py
+++ b/script/validators/source_validator.py
@@ -75,6 +75,8 @@ EXCLUDE_DIRECTORIES = [
 EXCLUDE_FILES = [
     # Django-generated files
     os.path.join(OTTERTUNE_DIR, 'server/website/manage.py'),
+    # file causing import error
+    os.path.join(OTTERTUNE_DIR, 'server/analysis/simulation.py'),
 ]
 
 CHECKSTYLE_JAR_PATH = os.path.join(OTTERTUNE_DIR,
diff --git a/server/analysis/simulation.py b/server/analysis/simulation.py
index dff650f..72b9681 100644
--- a/server/analysis/simulation.py
+++ b/server/analysis/simulation.py
@@ -5,12 +5,10 @@
 #
 
 
-import imp
 import random
 import os
 import sys
 try:
-    imp.find_module('matplotlib.pyplot')
     import matplotlib.pyplot as plt
 except (ModuleNotFoundError, ImportError):
     plt = None