frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 13:44:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Core unit test fixes for Python 3

Browse Source 
Summary: As title

Differential Revision: D5291327

fbshipit-source-id: 7dd9279c53ba55d3422c31973ffcec5705787fdf
This commit is contained in:
Thomas Dudziak
2017-06-23 13:06:20 -07:00
committed by Facebook Github Bot
parent ff914bf201
commit 342de07231
18 changed files with 167 additions and 131 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
caffe2/python/workspace_test.py
View File

@ -385,7 +385,6 @@ class TestCWorkspace(htu.HypothesisTestCase):
@given(name=st.text(), value=st.floats(min_value=-1, max_value=1.0))
def test_operator_run(self, name, value):
name = name.encode('ascii', 'ignore')
ws = workspace.C.Workspace()
op = core.CreateOperator(
"ConstantFill", [], [name], shape=[1], value=value)
@ -398,7 +397,6 @@ class TestCWorkspace(htu.HypothesisTestCase):
net_name=st.text(),
value=st.floats(min_value=-1, max_value=1.0))
def test_net_run(self, blob_name, net_name, value):
blob_name = blob_name.encode('ascii', 'ignore')
ws = workspace.C.Workspace()
net = core.Net(net_name)
net.ConstantFill([], [blob_name], shape=[1], value=value)
@ -413,7 +411,6 @@ class TestCWorkspace(htu.HypothesisTestCase):
plan_name=st.text(),
value=st.floats(min_value=-1, max_value=1.0))
def test_plan_run(self, blob_name, plan_name, net_name, value):
blob_name = blob_name.encode('ascii', 'ignore')
ws = workspace.C.Workspace()
plan = core.Plan(plan_name)
net = core.Net(net_name)
@ -431,7 +428,6 @@ class TestCWorkspace(htu.HypothesisTestCase):
net_name=st.text(),
value=st.floats(min_value=-1, max_value=1.0))
def test_net_create(self, blob_name, net_name, value):
blob_name = blob_name.encode('ascii', 'ignore')
ws = workspace.C.Workspace()
net = core.Net(net_name)
net.ConstantFill([], [blob_name], shape=[1], value=value)