[mlpack-svn] r15274 - mlpack/conf/jenkins-conf/benchmark

[fastlab-svn at coffeetalk-1.cc.gatech.edu]

Author: marcus
Date: 2013-06-20 17:19:42 -0400 (Thu, 20 Jun 2013)
New Revision: 15274

Modified:
   mlpack/conf/jenkins-conf/benchmark/benchmark.py
Log:
Add support for train and test dataset.

Modified: mlpack/conf/jenkins-conf/benchmark/benchmark.py
===================================================================
--- mlpack/conf/jenkins-conf/benchmark/benchmark.py	2013-06-20 21:16:28 UTC (rev 15273)
+++ mlpack/conf/jenkins-conf/benchmark/benchmark.py	2013-06-20 21:19:42 UTC (rev 15274)
@@ -63,7 +63,10 @@
 
 					row = ['-'] * 4;
 					# Get dataset name.
-					row[0] = os.path.splitext(os.path.basename(files))[0]
+					if  not isinstance(files, basestring):
+						row[0] = os.path.splitext(os.path.basename(files[0]))[0]	
+					else:
+						row[0] = os.path.splitext(os.path.basename(files))[0]	
 
 					Log.Info('Dataset: ' + row[0])