Repository: incubator-systemml Updated Branches: refs/heads/master 90f5cbba4 -> d6df0dbfb
[SYSTEMML-870] PYDML script type from file extension Closes #221. Project: http://git-wip-us.apache.org/repos/asf/incubator-systemml/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-systemml/commit/d6df0dbf Tree: http://git-wip-us.apache.org/repos/asf/incubator-systemml/tree/d6df0dbf Diff: http://git-wip-us.apache.org/repos/asf/incubator-systemml/diff/d6df0dbf Branch: refs/heads/master Commit: d6df0dbfbabb13ee8ad1052352dd122814b74b9f Parents: 90f5cbb Author: Deron Eriksson <[email protected]> Authored: Fri Aug 26 11:59:33 2016 -0700 Committer: Deron Eriksson <[email protected]> Committed: Fri Aug 26 11:59:33 2016 -0700 ---------------------------------------------------------------------- src/main/java/org/apache/sysml/api/DMLScript.java | 4 ++++ 1 file changed, 4 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-systemml/blob/d6df0dbf/src/main/java/org/apache/sysml/api/DMLScript.java ---------------------------------------------------------------------- diff --git a/src/main/java/org/apache/sysml/api/DMLScript.java b/src/main/java/org/apache/sysml/api/DMLScript.java index 3d76273..daf2144 100644 --- a/src/main/java/org/apache/sysml/api/DMLScript.java +++ b/src/main/java/org/apache/sysml/api/DMLScript.java @@ -35,6 +35,7 @@ import java.util.HashSet; import java.util.Map; import java.util.Scanner; +import org.apache.commons.lang.StringUtils; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.apache.hadoop.conf.Configuration; @@ -345,6 +346,9 @@ public class DMLScript setLoggingProperties( conf ); //Step 2: prepare script invocation + if (StringUtils.endsWithIgnoreCase(args[1], ".pydml")) { + parsePyDML = true; + } String dmlScriptStr = readDMLScript(args[0], args[1]); Map<String, String> argVals = createArgumentsMap(namedScriptArgs, scriptArgs);
