o qo™iŸã@sPdZddlmZddlmZddlmZGdd„dejƒZdedefd d „Z d S) akRoutines for nice string representations of cfg values. Some types (e.g. numpy arrays, jax dtypes, TensorFlow, etc) are often referred unambiguously by a name that differs from their qualified path. This library exposes the pretty-printing functionality that powers the codegen features to other string representations, such as graphviz or string serialization. é)ÚAny)Úlegacy_codegen)Úspecial_value_codegenc@s"eZdZdZdedefdd„ZdS)ÚLazyImportManageraImport manager for GraphViz & string formatting. We create an instance of the codegen's import manager on every method call, which effectively will alias imports like 'jax.numpy' to 'jnp', but will not create new import aliases when two modules of the same name are referenced. Ú module_nameÚreturncCst ¡}t |¡}| |¡S)zCReturns the usable name generated from an ephemeral import manager.)rÚ NamespaceÚ ImportManagerÚ add_by_name)ÚselfrÚ namespaceÚimport_manager©rú\/home/ubuntu/.local/lib/python3.10/site-packages/fiddle/_src/codegen/formatting_utilities.pyr &s  zLazyImportManager.add_by_nameN)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ústrr rrrrrsrÚvaluercCstt |tƒ¡ƒS)z?Returns the nicest eval-able string representation for `value`.)ÚreprrÚtransform_py_valuer)rrrrÚ pretty_print-s ÿrN) rÚtypingrÚfiddle._src.codegenrrÚImportManagerApirrrrrrrÚs