o
    ٷiK                     @   sZ   d dl Z d dlmZ ddlmZ ddlmZ d dlmZ ee	Z
e jG dd	 d	eZdS )
    N)List   )	CacheType   )BasicCacheConfig)init_loggerc                   @   sr   e Zd ZU ejZeed< dZeed< dZ	e
ed< dZe
ed< ejedZee ed	< dZeed
< defddZdS )DBPruneConfig
cache_typeFenable_dynamic_prune_thresholdNmax_dynamic_prune_thresholdg      ?#dynamic_prune_threshold_relax_ratio)default_factorynon_prune_block_idsforce_reduce_calibrator_vramreturnc                 C   sN   | j  d| j d| j d| j d| j dtd| j dtd| j d| j S )	N_FB_WIMr   MC_R)	r	   Fn_compute_blocksBn_compute_blocksmax_warmup_stepswarmup_intervalmaxmax_cached_stepsmax_continuous_cached_stepsresidual_diff_threshold)self r!   a/home/ubuntu/.local/lib/python3.10/site-packages/cache_dit/caching/cache_contexts/prune_config.pystrify3   s    

zDBPruneConfig.strify)__name__
__module____qualname__r   DBPruner	   __annotations__r
   boolr   floatr   dataclassesfieldlistr   r   intr   strr#   r!   r!   r!   r"   r      s   
 r   )r+   typingr   cache_typesr   cache_configr   cache_dit.loggerr   r$   logger	dataclassr   r!   r!   r!   r"   <module>   s    