From 3097163d22676254e3230185fa16e56db1a5a339 Mon Sep 17 00:00:00 2001
From: Samuel Musson <wppj21@ncc1.clients.dur.ac.uk>
Date: Wed, 26 Oct 2022 17:12:32 +0100
Subject: [PATCH] removed accidental IPython things

---
 src/molearn/molearn_trainer.py           | 4 ----
 src/molearn/openmm_loss/openmm_thread.py | 1 -
 2 files changed, 5 deletions(-)

diff --git a/src/molearn/molearn_trainer.py b/src/molearn/molearn_trainer.py
index 71360e2..5fd0d93 100644
--- a/src/molearn/molearn_trainer.py
+++ b/src/molearn/molearn_trainer.py
@@ -13,10 +13,6 @@
 from molearn.openmm_loss import openmm_energy
 import warnings
 from decimal import Decimal
-try:
-    from IPython import embed
-except ImportError:
-    print('IPython not install but also not strictly needed')
 
 class TrainingFailure(Exception):
     pass
diff --git a/src/molearn/openmm_loss/openmm_thread.py b/src/molearn/openmm_loss/openmm_thread.py
index 083f5f8..f75c4bb 100644
--- a/src/molearn/openmm_loss/openmm_thread.py
+++ b/src/molearn/openmm_loss/openmm_thread.py
@@ -11,7 +11,6 @@
     print('no plugin, wont be able to use openmm_loss')
 import torch
 from math import ceil
-from IPython import embed
 import numpy as np