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

[mboehm7]

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 <mboe...@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) {