o xi @sdZddlmZddlmZmZddlZddlmZzddl Z Wne y5Z z dZ eje de dZ [ wwdd dZ ddddZ ddddZdS) zKSupport for Pandas datatypes. May raise MissingDependencyError on import. ) annotations)AnyTypeIsN)errorszR`pandas` not installed >> @wandb_log(datasets=True) may not auto log your dataset!)warningdatarreturnTypeIs[pd.DataFrame]cCs t|tjS)z/Returns whether the data is a Pandas DataFrame.) isinstancepd DataFrame)rrZ/home/ubuntu/.local/lib/python3.10/site-packages/wandb/integration/metaflow/data_pandas.py is_dataframes rFnamestrrunwandb.Run | Nonetestingbool str | NonecCs6|rdS|sJtd|d||ddS)zLog a dependency on a DataFrame input. Args: name: Name of the input. run: The run to update. testing: True in unit tests. datasetszUsing artifact:  (Pandas DataFrame)z:latestN)wandbtermlog use_artifact)rrrrrr use_dataframes  r pd.DataFramecCs~|rdS|sJtj|dd}||dd}|j|ddWdn1s*wYtd |d ||dS) zLog a DataFrame output as an artifact. Args: name: The output's name. data: The output's value. run: The run to update. testing: True in unit tests. rdataset)typez.parquetwbpyarrow)engineNzLogging artifact: r)rArtifactnew_file to_parquetr log_artifact)rrrrartifactfrrrtrack_dataframe2s r*)rrr r )F)rrrrrrr r) rrrrrrrrr r)__doc__ __future__rtyping_extensionsrrrrpandasr ImportErrorerMissingDependencyErrorrrr*rrrrs$