| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253 |
- from __future__ import annotations
- from collections.abc import Mapping
- from typing import Any
- import wandb
- from ..sdk.lib import ipython
- class Attrs:
- def __init__(self, attrs: Mapping[str, Any]):
- self._attrs = dict(attrs)
- def snake_to_camel(self, string):
- camel = "".join([i.title() for i in string.split("_")])
- return camel[0].lower() + camel[1:]
- def display(self, height=420, hidden=False) -> bool:
- """Display this object in jupyter."""
- if wandb.run and wandb.run._settings.silent:
- return False
- if not ipython.in_jupyter():
- return False
- html = self.to_html(height, hidden)
- if html is None:
- wandb.termwarn("This object does not support `.display()`")
- return False
- try:
- from IPython import display
- except ImportError:
- wandb.termwarn(".display() only works in jupyter environments")
- return False
- display.display(display.HTML(html))
- return True
- def to_html(self, *args, **kwargs):
- return None
- def __getattr__(self, name):
- key = self.snake_to_camel(name)
- if key == "user":
- raise AttributeError
- if key in self._attrs:
- return self._attrs[key]
- elif name in self._attrs:
- return self._attrs[name]
- else:
- raise AttributeError(f"{repr(self)!r} object has no attribute {name!r}")
|
