File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -52,6 +52,7 @@ def do_POST(self):
52
52
request = json .loads (self .rfile .read (content_length ))
53
53
model_ref = request ["modelRef" ]
54
54
55
+
55
56
self .send_response (200 )
56
57
self .send_header ("Content-type" , "application/json" )
57
58
self .end_headers ()
@@ -88,7 +89,7 @@ def test_kaggle_resolver_long_url_succeeds(self):
88
89
def test_kaggle_resolver_short_url_succeeds (self ):
89
90
with create_test_server (KaggleJwtHandler ) as addr :
90
91
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 " )
92
93
self .assertEqual ([1 , 1 ], layer (test_inputs ).shape )
93
94
94
95
def test_kaggle_resolver_not_attached_throws (self ):
You can’t perform that action at this time.
0 commit comments