scripts/libraries: Fix postprocess arg default.

This commit is contained in:
Kwabena W. Agyeman 2025-10-24 20:44:02 -07:00
parent d038dc7ab8
commit 9fa764035b

View File

@ -33,7 +33,7 @@ from ml.preprocessing import Normalization
class Model(uml.Model):
def __init__(self, *args, **kwargs):
super().__init__(*args, kwargs.get("load_to_fb", False), kwargs.get("postprocess", False))
super().__init__(*args, kwargs.get("load_to_fb", False), kwargs.get("postprocess", None))
try:
path = args[0].split(".")[0] + ".txt"
self.labels = [line.rstrip('\n') for line in open(path, "r")]