You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@systemds.apache.org by mb...@apache.org on 2020/11/09 10:57:34 UTC

[systemds] branch master updated: [MINOR] Fix ML context tests (lineage cleanup on object cleanup)

This is an automated email from the ASF dual-hosted git repository.

mboehm7 pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/systemds.git


The following commit(s) were added to refs/heads/master by this push:
     new 997ba10  [MINOR] Fix ML context tests (lineage cleanup on object cleanup)
997ba10 is described below

commit 997ba10ac59414685a76f97739126eac1f415e97
Author: Matthias Boehm <mb...@gmail.com>
AuthorDate: Mon Nov 9 11:57:07 2020 +0100

    [MINOR] Fix ML context tests (lineage cleanup on object cleanup)
---
 .../org/apache/sysds/runtime/controlprogram/caching/CacheableData.java   | 1 +
 .../sysds/test/functions/mlcontext/MLContextScratchCleanupTest.java      | 1 -
 2 files changed, 1 insertion(+), 1 deletion(-)

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 @@ public abstract class CacheableData<T extends CacheBlock> extends Data
 		// 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 @@ public class MLContextScratchCleanupTest extends AutomatedTestBase
 			
 			Script script2 = dmlFromFile(dml2).in("X", X).out("z");
 			String z = ml.execute(script2).getString("z");
-			
 			System.out.println(z);
 		}
 		catch(Exception ex) {