From 051e1e1b454a73957dbca955abf8ef6f9bd4a566 Mon Sep 17 00:00:00 2001
From: Yu Shan <shan.yu@intel.com>
Date: Wed, 12 Aug 2020 15:09:49 +0800
Subject: [PATCH] Add horovod estimator save/load/get_models/shutdown (#2702)

* add get_model/save/load/shutdown for pytorch horovod estimator

* change args

* remove condition

* change name

* remove get multiple models

* add pytorch estimator ut and disable for now

* meet comments
---
 python/orca/src/test/dev/run-pytests-ray | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/orca/src/test/dev/run-pytests-ray b/python/orca/src/test/dev/run-pytests-ray
index 299076b4919..8f0dcee05ce 100755
--- a/python/orca/src/test/dev/run-pytests-ray
+++ b/python/orca/src/test/dev/run-pytests-ray
@@ -34,7 +34,8 @@ then
 fi
 
 echo "Running orca learn ray tests"
-python -m pytest -v ../test/zoo/orca/learn/ray
+python -m pytest -v ../test/zoo/orca/learn/ray \
+       --ignore=../test/zoo/orca/learn/ray/pytorch/test_pytorch_estimator.py
 exit_status_2=$?
 if [ $exit_status_2 -ne 0 ];
 then