We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6b0e57e commit c778140Copy full SHA for c778140
elephant/test/test_asset.py
@@ -842,8 +842,13 @@ def cleanUpClass(cls):
842
# Restore environment flags
843
if cls.use_cuda:
844
os.environ["ELEPHANT_USE_CUDA"] = cls.use_cuda
845
+ else:
846
+ os.environ.pop("ELEPHANT_USE_CUDA")
847
+
848
if cls.use_opencl:
849
os.environ["ELEPHANT_USE_OPENCL"] = cls.use_opencl
850
851
+ os.environ.pop("ELEPHANT_USE_OPENCL")
852
853
854
@unittest.skipUnless(HAVE_SKLEARN, 'requires sklearn')
0 commit comments