From 8ecccf2aa25527bd469fb24c87df1ed0bc560bd5 Mon Sep 17 00:00:00 2001
From: Timothy Poon <timothy.poon@diamond.ac.uk>
Date: Tue, 10 Sep 2024 13:46:19 +0100
Subject: [PATCH] Add missing import in LBFGS cupy

---
 ptypy/custom/LBFGS_cupy.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/ptypy/custom/LBFGS_cupy.py b/ptypy/custom/LBFGS_cupy.py
index 88a769d34..564a977b3 100644
--- a/ptypy/custom/LBFGS_cupy.py
+++ b/ptypy/custom/LBFGS_cupy.py
@@ -12,10 +12,7 @@
     :license: GPLv2, see LICENSE for details.
 """
 import numpy as np
-# from pycuda import gpuarray
-# import pycuda.driver as cuda
-# import pycuda.cumath
-# from pycuda.tools import DeviceMemoryPool
+import cupy as cp
 
 from ptypy.engines import register
 from ptypy.custom.LBFGS_serial import LBFGS_serial