Skip to content

Commit 19a3269

Browse files
committed
fix ut
1 parent 8e9e8e8 commit 19a3269

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

dlrover/python/tests/test_servicer.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1111
# See the License for the specific language governing permissions and
1212
# limitations under the License.
13-
13+
import os
1414
import time
1515
import unittest
1616
from unittest import mock
@@ -85,6 +85,9 @@ def setUp(self) -> None:
8585
sync_service=sync_service,
8686
)
8787

88+
def tearDown(self) -> None:
89+
os.environ.clear()
90+
8891
def test_query_running_nodes(self):
8992
request = elastic_training_pb2.Message()
9093
message = grpc.RunningNodesRequest()

0 commit comments

Comments
 (0)