Skip to content

Commit e6cd909

Browse files
authored
Fix docstring of Python interpreter (#677)
1 parent c5d7e32 commit e6cd909

File tree

2 files changed

+10
-15
lines changed

2 files changed

+10
-15
lines changed

larq_compute_engine/tflite/python/interpreter.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,20 @@ class Interpreter(InterpreterBase):
1919
interpreter.predict(input_data, verbose=1)
2020
```
2121
22-
See the base class `InterpreterBase` for the full interface.
23-
2422
# Arguments
2523
flatbuffer_model: A serialized Larq Compute Engine model in the flatbuffer format.
2624
num_threads: The number of threads used by the interpreter.
2725
use_reference_bconv: When True, uses the reference implementation of LceBconv2d.
26+
27+
# Attributes
28+
input_types: Returns a list of input types.
29+
input_shapes: Returns a list of input shapes.
30+
input_scales: Returns a list of input scales.
31+
input_zero_points: Returns a list of input zero points.
32+
output_types: Returns a list of output types.
33+
output_shapes: Returns a list of output shapes.
34+
output_scales: Returns a list of input scales.
35+
output_zero_points: Returns a list of input zero points.
2836
"""
2937

3038
def __init__(

larq_compute_engine/tflite/python/interpreter_base.py

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -27,19 +27,6 @@ def data_generator(x: Union[Data, Iterator[Data]]) -> Iterator[List[np.ndarray]]
2727

2828

2929
class InterpreterBase:
30-
"""Interpreter interface for Larq Compute Engine Models.
31-
32-
# Attributes
33-
input_types: Returns a list of input types.
34-
input_shapes: Returns a list of input shapes.
35-
input_scales: Returns a list of input scales.
36-
input_zero_points: Returns a list of input zero points.
37-
output_types: Returns a list of output types.
38-
output_shapes: Returns a list of output shapes.
39-
output_scales: Returns a list of input scales.
40-
output_zero_points: Returns a list of input zero points.
41-
"""
42-
4330
def __init__(self, interpreter):
4431
self.interpreter = interpreter
4532

0 commit comments

Comments
 (0)