j \ddlZddlZddlZddlmZddlmZmZmZm Z m Z m Z GddZ y)N)Path)AnyCallableDictOptionalTupleUnioncleZdZUdZdZeed<ejZ ejed<e ed<e ed<e e ed<dd e d e fd Z d ed ee ede fffdZde d e fdZde d ee ede fffdZdZej,dZdZdedefdZdeeeffdZdeeefd dfdZdZddZ y) ShimaZDefine a basic interface for external models. Users can create subclasses of 'shim' to wrap external libraries. We provide shims for PyTorch. The Thinc Model class treats Shim objects as a sort of special type of sublayer: it knows they're not actual Thinc Model instances, but it also knows to talk to the shim instances when doing things like using transferring between devices, loading in parameters, optimization. It also knows Shim objects need to be serialized and deserialized with to/from bytes/disk, rather than expecting that they'll be msgpack-serializable. r global_idglobal_id_lockcfg_model _optimizerNmodel optimizerctj5txjdz c_tj|_ddd| t |ni|_||_||_y#1swY,xYw)N)r r r iddictrrr)selfrconfigrs Z/home/spazio/kingdom_calls/work5/transenv/lib/python3.12/site-packages/thinc/shims/shim.py__init__z Shim.__init__s[   % NNa NnnDG %$*#54<2 #  % %s /A++A4is_trainreturn.ctNNotImplementedError)rinputsrs r__call__z Shim.__call__#!!fwd_argsc ||d\}}|S)NFr)rr%Ybackprops rpredictz Shim.predict&s8e4 8r$c||dS)NTr'r()rr%s r begin_updatezShim.begin_update*sHt,,r$ctrr)rrs r finish_updatezShim.finish_update-r#r$c#Kdywrr()rparamss r use_paramszShim.use_params0s  sc,tj|Sr)copydeepcopyrs rr4z Shim.copy4s}}T""r$ device_type device_idctrr)rr7r8s r to_devicezShim.to_device7r#r$pathc|j}t|tr t|n|}|j d5}|j |dddy#1swYyxYw)Nwb)to_bytes isinstancestrropenwrite)rr; bytes_datafile_s rto_diskz Shim.to_disk:sO]]_ 'c2tDz YYt_ $ KK # $ $ $s AA#ct|tr t|n|}|jd5}|j }ddd|j S#1swYxYw)Nrb)r?r@rrAread from_bytes)rr;rDrCs r from_diskzShim.from_disk@sR'c2tDz YYt_ &J &z** & &s AA"ctrrr6s rr>z Shim.to_bytesFr#r$ctrr)rdatas rrIzShim.from_bytesIr#r$)NN)rr )!__name__ __module__ __qualname____doc__r int__annotations__ threadingLockr rrrrboolrrr"r+r-r/ contextlibcontextmanagerr2r4r@r:r rrErJr>rIr(r$rr r s" Is%3Y^^%5NINN5 I K $c$3$""%Xc3h=O8O2P"-S-U3c8J3J-K-"#"S"S"$E#t),$ +eCI.+6+ ""r$r ) rWr4rTpathlibrtypingrrrrrr r r(r$rr[s# >>B"B"r$