Skip to content

Commit cf78793

Browse files
committed
Fix test
1 parent 2663eec commit cf78793

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/test_kaggle_module_resolver.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ def do_POST(self):
5252
request = json.loads(self.rfile.read(content_length))
5353
model_ref = request["modelRef"]
5454

55+
5556
self.send_response(200)
5657
self.send_header("Content-type", "application/json")
5758
self.end_headers()
@@ -88,7 +89,7 @@ def test_kaggle_resolver_long_url_succeeds(self):
8889
def test_kaggle_resolver_short_url_succeeds(self):
8990
with create_test_server(KaggleJwtHandler) as addr:
9091
test_inputs = tf.ones([1,4])
91-
layer = hub.KerasLayer("https://kaggle.com/models/foo/foomodule/TensorFlow2/barvar/2")
92+
layer = hub.KerasLayer("https://kaggle.com/models/bar/barmodule/pyTorch/barvar/1")
9293
self.assertEqual([1, 1], layer(test_inputs).shape)
9394

9495
def test_kaggle_resolver_not_attached_throws(self):

0 commit comments

Comments
 (0)