Update src/tofromdatavalues.jl

Co-authored-by: Jacob Quinn <[email protected]>
JuliaData · Apr 10, 2022 · 3ddc401 · 3ddc401
1 parent 06d0ed5
commit 3ddc401
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/src/tofromdatavalues.jl b/src/tofromdatavalues.jl
@@ -93,15 +93,7 @@ IteratorInterfaceExtensions.getiterator(x::MyTable) = Tables.datavaluerows(x)
 ```
 """
 function datavaluerows(x)
-    x2 = if Tables.columnaccess(x)
-        Tables.columntable(x)
-    elseif Tables.rowaccess(x)
-        x
-    else
-        error("Invalid input table, must either declare to have a Tables.rows or Tables.columns method.")
-    end
-
-    r = Tables.rows(x2)
+    r = Tables.rows(Tables.columnaccess(x) ? Tables.columntable(x) : x)
     s = Tables.schema(r)
     s === nothing && error("Schemaless sources cannot be passed to datavaluerows.")
     return DataValueRowIterator{datavaluenamedtuple(s), typeof(s), typeof(r)}(r)