feat: add metaclass to auto wrap a kw_only dataclass

SuperDuperDB / superduperdb

🔮 SuperDuperDB: Bring AI to your database! Build, deploy and manage any AI application directly with your existing data infrastructure, without moving your data. Including streaming inference, scalable model training and vector search.

https://superduperdb.com

Apache License 2.0

4.55k stars 445 forks source link

feat: add metaclass to auto wrap a kw_only dataclass #2144

Closed blythed closed 1 month ago

blythed commented 1 month ago

Description

Solves #2129
Removes @merge_docstrings in favour of metaclass
Adds @ensure_initialized to a metaclass for Model
Renames predict -> predict_batches and predict_one -> predict
Adds a default behaviour for predict_batches