o
    }oi                     @   sL   d dl mZ d dlmZ d dlmZmZ d dlmZ G dd dee Z	dS )    )Path)Any)dumpload)Artifactc                   @   sF   e Zd ZdededefddZdedefddZdedefd	d
ZdS )PickleArtifactabsolute_dirrelative_dirreturnc                 C   sL   |  |}tt|| d}tt| W d    |S 1 sw   Y  |S )Nwb)	file_pathopenr   r   value)selfinstancer   r	   relative_filef r   U/home/ubuntu/.local/lib/python3.10/site-packages/nemo/lightning/io/artifact/pickle.pyr      s   

zPickleArtifact.dumppathc                 C   s<   t | |d}t|W  d    S 1 sw   Y  d S )Nrb)r   r   r   )r   r   r   r   r   r   r      s   $zPickleArtifact.loadc                 C   s   || j  d S )Nz.pkl)attr)r   r   r   r   r   r   #   s   zPickleArtifact.file_pathN)__name__
__module____qualname__r   r   r   r   r   r   r   r   r   r      s    r   N)
pathlibr   typingr   cloudpickler   r   nemo.lightning.io.artifact.baser   r   r   r   r   r   <module>   s
   