Merge pull request #5 from josevalim/jv-skip-resnet50

Exclude resnet5 tests if model is not available

Author: mortont

lib/ortex.ex

@@ -13,9 +13,9 @@ defmodule Ortex do
 
   ## Examples
 
-      iex> Ortex.load("./models/resnet50.onnx")
-      iex> Ortex.load("./models/resnet50.onnx", [:cuda, :cpu])
-      iex> Ortex.load("./models/resnet50.onnx", [:cpu], 0)
+      iex> Ortex.load("./models/tinymodel.onnx")
+      iex> Ortex.load("./models/tinymodel.onnx", [:cuda, :cpu])
+      iex> Ortex.load("./models/tinymodel.onnx", [:cpu], 0)
 
   """
   defdelegate load(path, eps \\ [:cpu], opt \\ 3), to: Ortex.Model
@@ -33,11 +33,6 @@ defmodule Ortex do
 
   ## Examples
 
-      iex> model = Ortex.load("./models/resnet50.onnx")
-      iex> {%Nx.Tensor{shape: {1, 1000}}} = Ortex.run(
-      ...>    model, Nx.broadcast(0.0, {1, 3, 224, 224}))
-
-
       iex> model = Ortex.load("./models/tinymodel.onnx")
       iex> {%Nx.Tensor{shape: {1, 10}},
       ...>  %Nx.Tensor{shape: {1, 10}},
@@ -46,6 +41,7 @@ defmodule Ortex do
       ...>      Nx.broadcast(0, {1, 100}) |> Nx.as_type(:s32),
       ...>      Nx.broadcast(0, {1, 100}) |> Nx.as_type(:f32)
       ...>    })
+
   """
   defdelegate run(model, tensors), to: Ortex.Model
 end

test/ortex_test.exs

@@ -2,6 +2,7 @@ defmodule OrtexTest do
   use ExUnit.Case
   doctest Ortex
 
+  @tag :resnet50
   test "resnet50" do
     model = Ortex.load("./models/resnet50.onnx")
 
@@ -12,6 +13,7 @@ defmodule OrtexTest do
     assert argmax == Nx.tensor([499])
   end
 
+  @tag :resnet50
   test "Nx.Serving with resnet50" do
     model = Ortex.load("./models/resnet50.onnx")
 

test/test_helper.exs

@@ -1 +1,16 @@
-ExUnit.start()
+exclude =
+  if File.exists?("models/resnet50.onnx") do
+    []
+  else
+    IO.warn(
+      """
+      skipping resnet50 tests because model is not available.
+      Run python/export_resnet.py before for a complete test suite\
+      """,
+      []
+    )
+
+    [:resnet50]
+  end
+
+ExUnit.start(exclude: exclude)