From ecb0af316ebd48121d5ef48e4252ce253317ff82 Mon Sep 17 00:00:00 2001
From: Gene <liujd@lamda.nju.edu.cn>
Date: Fri, 31 Mar 2023 22:08:01 +0800
Subject: [PATCH] [ENH] Modify Learnware

---
 learnware/learnware/base.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/learnware/learnware/base.py b/learnware/learnware/base.py
index 76482e1..eb2aaea 100644
--- a/learnware/learnware/base.py
+++ b/learnware/learnware/base.py
@@ -1,9 +1,10 @@
+import os
 import numpy as np
 from typing import Union
-from ..specification import Specification
+
+from ..specification import Specification, BaseStatSpecification
 from ..model import BaseModel
 from ..utils import get_module_by_module_path
-import os
 
 
 class Learnware:
@@ -47,13 +48,16 @@ class Learnware:
             raise TypeError("model must be BaseModel or dict")
 
     def get_model(self) -> BaseModel:
-        pass
+        return self.model
 
     def get_specification(self) -> Specification:
-        pass
+        return self.specification
 
     def get_info(self):
-        pass
+        return self.desc
+    
+    def update_stat_spec(self, name, new_stat_spec: BaseStatSpecification):
+        self.specification.update_stat_spec(name, new_stat_spec)
 
     def update(self):
         # Empty Interface.