o ^_h @sddlZddlZddlZddlZddlZddlmZddlZddlm Z gdZ ej e dZ eeejdZddZd d Zd eed eeeffd dZded dfddZdedBdedBd eedBedBffddZddZdS)N)Path)HF_HOME)projectrun timestampsteptimetrackioztrackio_logo.pngcCsDgd}gd}t|}t|}tdd}|d|d|S)zB Generates a random, readable name like "dainty-sunset-1" )ZZdaintyZbraveZcalmeagerfancyZgentleZhappyZjollykindlivelyZmerryniceZproudquickZsillyZtidyZwittyZzealousbrightshyboldZcleverZdaringZelegantZfaithfulZgracefulZhonestZ inventiveZjovialZkeenZluckyZmodestZnobleZ optimisticZpatientZquirkyZ resourcefulZsincereZ thoughtfulZupbeatZvaliantZwarmZyouthfulZzestyZ adventurousZbreezyZcheerfulZ delightfulZ energeticZfearlessZgladZhopefulZ imaginativeZjoyfulZkindlyZluminousZ mysteriousZneatoutgoingZplayfulZradiantZspiritedZtranquiluniqueZvividZwiseZzanyZartfulZbubblyZcharmingZdazzlingZearnestZfestiveZ gentlemanlyZheartyZintrepidZjubilantZknightlyr ZmagneticZnimbleZorderlyZpeacefulz quick-wittedZrobustZsturdyZtrustyZ upstandingZvibrantZ whimsical)ZsunsetZforestZrivermountainZbreezeZmeadowoceanZvalleyskyfieldcloudstarZrainleafstoneZflowerbirdtreewavetrailZislanddesertZhillZlakeZpondZgroveZcanyonZreefZbayZpeakZgladeZmarshZcliffZduneZspringZbrookZcaveplainZridgeZwoodblossomZpetalrootbranchseedZacornZpineZwillowZcedarelmZfalconeagleZsparrowZrobinowlZfinchZheronZcraneduckswanfoxwolfbeardeerZmooseotterZbeaverlynxharebadger butterflybeeantbeetleZ dragonflyZfireflyZladybugZmothspiderwormcoralZkelpshellZpebbleZboulderZcobbleZsandZwaveletZtidecurrentc-)randomchoicerandint)Z adjectivesZnounsZ adjectivenounnumberrF5/Users/nouamanetazi/projects/trackio/trackio/utils.pygenerate_readable_names \ \  rHc CsLtttdtjj}|rdSz tdqttfy%t dYdSw)Nps1Tg?z:Keyboard interruption in main thread... closing dashboard.) boolgetattrsysflags interactiversleepKeyboardInterruptOSErrorprint)Z in_notebookrFrFrGblock_except_in_notebooks rScolumnsreturncCsi}t}|D]8}tdd|}|r|ddndt|}|}d}||vr6|d|}|d7}||vs'|||<||q|S)z Simplifies column names to first 10 alphanumeric or "/" characters with unique suffixes. Args: columns: List of original column names Returns: Dictionary mapping original column names to simplified names z [^a-zA-Z0-9/]N col_r>_)setresublenadd)rTZsimplified_names used_namescol alphanumeric base_name final_namesuffixrFrFrGsimplify_column_namess  rercCsDd}d}d}tdt||d|d|td|dd S) z Prints instructions for viewing the Trackio dashboard. Args: project: The name of the project to show dashboard for. zzzz-* View dashboard by running in your terminal:ztrackio show --project ""z1* or by running in Python: trackio.show(project="z")N)rR)rYELLOWZBOLDRESETrFrFrGprint_dashboard_instructionss rispace_id dataset_idcCsv|durd|vrtd}|d|}|dur*d|vr*td}|d|}|dur7|dur7|d}||fS)N/nameZ_dataset)huggingface_hubwhoami)rjrkusernamerFrFrG preprocess_space_and_dataset_ids s   rqccs"d\}} |V|||}}q)z6Generator for Fibonacci backoff: 1, 1, 2, 3, 5, 8, ...)r>r>rF)abrFrFrGfibos rt)osrAr[rLrpathlibrrnZhuggingface_hub.constantsrZ RESERVED_KEYSpathjoin TRACKIO_DIRstr__file__parentjoinpathTRACKIO_LOGO_PATHrHrSlistdictrerituplerqrtrFrFrFrGs0  C