Skip to content

Commit f141bf0

Browse files
fix more display issues
1 parent 4b884a4 commit f141bf0

File tree

3 files changed

+11
-6
lines changed

3 files changed

+11
-6
lines changed

mindee/parsing/v2/base_field.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ def __init__(self, raw_response: StringDict, indent_level: int = 0):
8484
def __str__(self) -> str:
8585
out_str = ""
8686
for item in self.items:
87-
out_str += f"* {str(item)[2:] if item else ''}"
87+
out_str += f"*{str(item)[2:] if item else ''}"
8888
return "\n" + out_str if out_str else ""
8989

9090

mindee/parsing/v2/inference.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,9 @@ def __str__(self) -> str:
2828
return (
2929
f"Inference\n"
3030
f"#########\n"
31-
f":Model: {self.model.id}\n"
32-
f":File:\n"
33-
f" :Name: {self.file.name}\n"
34-
f" :Alias: {self.file.alias}\n\n"
31+
f":model: {self.model.id}\n"
32+
":file:\n"
33+
f"{self.file.__str__(1)}"
3534
f"Result\n"
3635
f"======\n"
3736
f"{self.result}\n"
Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
from typing import Optional
2+
13
from mindee.parsing.common.string_dict import StringDict
24

35

@@ -6,9 +8,13 @@ class InferenceFile:
68

79
name: str
810
"""Name of the file."""
9-
alais: str
11+
alias: Optional[str]
1012
"""Alias of the file."""
1113

1214
def __init__(self, raw_response: StringDict) -> None:
1315
self.name = raw_response["name"]
1416
self.alias = raw_response["alias"]
17+
18+
def __str__(self, indent: int = 0) -> str:
19+
return f"{' ' * indent}:name: {self.name}\n"\
20+
f"{' ' * indent}:alias: {self.alias if self.alias else ''}\n"

0 commit comments

Comments
 (0)