Skip to content

Commit c8f36d1

Browse files
committed
[DEBUG] Add mode debug output to check stuff happening in workflow
1 parent 00c289d commit c8f36d1

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

medcat-v2/tests/components/addons/meta_cat/test_bert_meta_cat.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,10 @@
2121
def assert_tries_network():
2222
real_socket = socket.socket
2323
calls = []
24+
print(f"=== DEBUG: Network call interception started on thread {threading.get_ident()} ===")
2425

2526
def guard(*args, **kwargs):
27+
print(f"=== DEBUG: Network call intercepted on thread {threading.get_ident()} ===")
2628
calls.append((len(args), len(kwargs)))
2729
raise OSError("Network disabled for test")
2830

@@ -62,6 +64,7 @@ def replacement_method(*args, **kwargs):
6264
yield
6365
finally:
6466
transformers.BertModel.from_pretrained = orig_from_pretrained
67+
print(f"=== DEBUG: from_pretrained called {len(method_calls)} times ===")
6568
assert method_calls, "BertModel.from_pretrained should be called"
6669

6770

0 commit comments

Comments
 (0)