Use shape in fetch

versilov · Mar 12, 2019 · 45e0786 · 45e0786
1 parent bb28701
commit 45e0786
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/matrex.ex b/lib/matrex.ex
@@ -528,7 +528,7 @@ defmodule Matrex do
   def fetch(%Matrex{shape: {rows, _}}, :rows), do: {:ok, rows}
   def fetch(%Matrex{shape: {_, cols}}, :cols), do: {:ok, cols}
   def fetch(%Matrex{shape: {_, cols}}, :columns), do: {:ok, cols}
-  def fetch(%Matrex{shape: {rows, cols}}, :size), do: {:ok, {rows, cols}}
+  def fetch(%Matrex{shape: shape}, :size), do: {:ok, shape}
   def fetch(matrex, :sum), do: {:ok, sum(matrex)}
   def fetch(matrex, :max), do: {:ok, max(matrex)}
   def fetch(matrex, :min), do: {:ok, min(matrex)}