Dhairyashil Ghatage commited on
Commit
251429f
·
1 Parent(s): a9416f6

add base model

Browse files
Files changed (1) hide show
  1. models/base.py +15 -0
models/base.py ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import inspect
2
+ from dataclasses import dataclass
3
+
4
+
5
+ @dataclass
6
+ class BaseModelArgs:
7
+ @classmethod
8
+ def from_dict(cls, params):
9
+ return cls(
10
+ **{
11
+ k: v
12
+ for k, v in params.items()
13
+ if k in inspect.signature(cls).parameters
14
+ }
15
+ )