Author: srowen
Date: Tue Jan 22 11:41:33 2013
New Revision: 1436840
URL: http://svn.apache.org/viewvc?rev=1436840&view=rev
Log:
MAHOUT-1142 support single file input
Modified:
mahout/trunk/core/src/main/java/org/apache/mahout/math/hadoop/DistributedRowMatrix.java
Modified:
mahout/trunk/core/src/main/java/org/apache/mahout/math/hadoop/DistributedRowMatrix.java
URL:
http://svn.apache.org/viewvc/mahout/trunk/core/src/main/java/org/apache/mahout/math/hadoop/DistributedRowMatrix.java?rev=1436840&r1=1436839&r2=1436840&view=diff
==============================================================================
---
mahout/trunk/core/src/main/java/org/apache/mahout/math/hadoop/DistributedRowMatrix.java
(original)
+++
mahout/trunk/core/src/main/java/org/apache/mahout/math/hadoop/DistributedRowMatrix.java
Tue Jan 22 11:41:33 2013
@@ -134,8 +134,12 @@ public class DistributedRowMatrix implem
@Override
public Iterator<MatrixSlice> iterateAll() {
try {
+ Path pathPattern = rowPath;
+ if (FileSystem.get(conf).getFileStatus(rowPath).isDir()) {
+ pathPattern = new Path(rowPath, "*");
+ }
return Iterators.transform(
- new SequenceFileDirIterator<IntWritable,VectorWritable>(new
Path(rowPath, "*"),
+ new SequenceFileDirIterator<IntWritable,VectorWritable>(pathPattern,
PathType.GLOB,
PathFilters.logsCRCFilter(),
null,
