| commit | 997ba10ac59414685a76f97739126eac1f415e97 | [log] [tgz] |
|---|---|---|
| author | Matthias Boehm <mboehm7@gmail.com> | Mon Nov 09 11:57:07 2020 +0100 |
| committer | Matthias Boehm <mboehm7@gmail.com> | Mon Nov 09 11:57:07 2020 +0100 |
| tree | 97702fa110dd11891e0c57b141659d096c9a7df4 | |
| parent | f2820a1f8dca214a82b66c34819b7eeb6dba7ba6 [diff] |
[MINOR] Fix ML context tests (lineage cleanup on object cleanup)
diff --git a/src/main/java/org/apache/sysds/runtime/controlprogram/caching/CacheableData.java b/src/main/java/org/apache/sysds/runtime/controlprogram/caching/CacheableData.java index 9b465a2..e3b4597 100644 --- a/src/main/java/org/apache/sysds/runtime/controlprogram/caching/CacheableData.java +++ b/src/main/java/org/apache/sysds/runtime/controlprogram/caching/CacheableData.java
@@ -711,6 +711,7 @@ // clear the in-memory data _data = null; clearCache(); + setCacheLineage(null); // clear rdd/broadcast back refs if( _rddHandle != null )
diff --git a/src/test/java/org/apache/sysds/test/functions/mlcontext/MLContextScratchCleanupTest.java b/src/test/java/org/apache/sysds/test/functions/mlcontext/MLContextScratchCleanupTest.java index 519bcd3..28a6543 100644 --- a/src/test/java/org/apache/sysds/test/functions/mlcontext/MLContextScratchCleanupTest.java +++ b/src/test/java/org/apache/sysds/test/functions/mlcontext/MLContextScratchCleanupTest.java
@@ -103,7 +103,6 @@ Script script2 = dmlFromFile(dml2).in("X", X).out("z"); String z = ml.execute(script2).getString("z"); - System.out.println(z); } catch(Exception ex) {