Spaces:
No application file
No application file
File size: 2,001 Bytes
8cb4f3b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
class MockLoader:
def __init__(self, *args, **kwargs):
"""Only print stuff"""
print("MockLoader initialized, args/kwargs {} {}".format(args, kwargs))
def tokenize(self, inputs, **kwargs):
print("MockLoader tokenize called, inputs/kwargs {} {}".format(inputs, kwargs))
return inputs
def detokenize(self, inputs, **kwargs):
print("MockLoader detokenize called, inputs/kwargs {} {}".format(inputs, kwargs))
return inputs
def reverse_lookup(self, inputs, **kwargs):
print("MockLoader reverse_lookup called, inputs/kwargs {} {}".format(inputs, kwargs))
return inputs
def lookup(self, inputs, **kwargs):
print("MockLoader lookup called, inputs/kwargs {} {}".format(inputs, kwargs))
return inputs
def embed(self, inputs, **kwargs):
print("MockLoader embed called, inputs/kwargs {} {}".format(inputs, kwargs))
return inputs
class MockEncoder:
def __init__(self, *args, **kwargs):
"""Only print stuff"""
print("MockEncoder initialized, args/kwargs {} {}".format(args, kwargs))
def encode(self, inputs, **kwargs):
print("MockEncoder encode called, inputs/kwargs {} {}".format(inputs, kwargs))
return inputs
def __call__(self, inputs, num_layers=3, **kwargs):
print("MockEncoder __call__ called, inputs/num_layers/kwargs {} {} {}".format(inputs, num_layers, kwargs))
for i in range(num_layers):
inputs = encode(inputs, **kwargs)
return inputs
class MockDecoder:
def __init__(self, *args, **kwargs):
"""Only print stuff"""
print("MockDecoder initialized, args/kwargs {} {}".format(args, kwargs))
def decode(self, inputs, **kwargs):
print("MockDecoder decode called, inputs/kwargs {} {}".format(inputs, kwargs))
return inputs
def __call__(self, inputs, num_layers=3, **kwargs):
print("MockDecoder __call__ called, inputs/num_layers/kwargs {} {} {}".format(inputs, num_layers, kwargs))
for i in range(num_layers):
inputs = decode(inputs, **kwargs)
return inputs
|