MCPcopy Create free account
hub / github.com/PAIR-code/lit / __init__

Method __init__

lit_nlp/app.py:916–1086  ·  view source on GitHub ↗
(
      self,
      models: lit_model.ModelMap,
      datasets: Mapping[str, lit_dataset.Dataset],
      generators: Optional[Mapping[str, lit_components.Generator]] = None,
      interpreters: Optional[Mapping[str, lit_components.Interpreter]] = None,
      metrics: Optional[Mapping[str, lit_components.Metrics]] = None,
      annotators: Optional[list[lit_components.Annotator]] = None,
      layouts: Optional[layout.LitComponentLayouts] = None,
      dataset_loaders: Optional[DatasetLoadersMap] = None,
      model_loaders: Optional[ModelLoadersMap] = None,
      # General server config; see server_flags.py.
      data_dir: Optional[str] = None,
      warm_start: float = 0.0,
      warm_start_progress_indicator: Optional[
          ProgressIndicator
      ] = tqdm.tqdm,  # not in server_flags
      warm_projections: bool = False,
      client_root: Optional[str] = None,
      demo_mode: bool = False,
      default_layout: Optional[str] = None,
      canonical_url: Optional[str] = None,
      page_title: Optional[str] = None,
      development_demo: bool = False,
      inline_doc: Optional[str] = None,
      onboard_start_doc: Optional[str] = None,
      onboard_end_doc: Optional[str] = None,
      sync_state: bool = False,  # notebook-only; not in server_flags
      validate: Optional[flag_helpers.ValidationMode] = None,
      report_all: bool = False,
      enforce_dataset_fields_required: bool = False,
      strict_cache_id_validation: bool = False,
  )

Source from the content-addressed store, hash-verified

914 return _handler
915
916 def __init__(
917 self,
918 models: lit_model.ModelMap,
919 datasets: Mapping[str, lit_dataset.Dataset],
920 generators: Optional[Mapping[str, lit_components.Generator]] = None,
921 interpreters: Optional[Mapping[str, lit_components.Interpreter]] = None,
922 metrics: Optional[Mapping[str, lit_components.Metrics]] = None,
923 annotators: Optional[list[lit_components.Annotator]] = None,
924 layouts: Optional[layout.LitComponentLayouts] = None,
925 dataset_loaders: Optional[DatasetLoadersMap] = None,
926 model_loaders: Optional[ModelLoadersMap] = None,
927 # General server config; see server_flags.py.
928 data_dir: Optional[str] = None,
929 warm_start: float = 0.0,
930 warm_start_progress_indicator: Optional[
931 ProgressIndicator
932 ] = tqdm.tqdm, # not in server_flags
933 warm_projections: bool = False,
934 client_root: Optional[str] = None,
935 demo_mode: bool = False,
936 default_layout: Optional[str] = None,
937 canonical_url: Optional[str] = None,
938 page_title: Optional[str] = None,
939 development_demo: bool = False,
940 inline_doc: Optional[str] = None,
941 onboard_start_doc: Optional[str] = None,
942 onboard_end_doc: Optional[str] = None,
943 sync_state: bool = False, # notebook-only; not in server_flags
944 validate: Optional[flag_helpers.ValidationMode] = None,
945 report_all: bool = False,
946 enforce_dataset_fields_required: bool = False,
947 strict_cache_id_validation: bool = False,
948 ):
949 if client_root is None:
950 raise ValueError('client_root must be set on application')
951 self._demo_mode = demo_mode
952 self._development_demo = development_demo
953 self._default_layout = default_layout
954 self._canonical_url = canonical_url
955 self._page_title = page_title
956 self._inline_doc = inline_doc
957 self._onboard_start_doc = onboard_start_doc
958 self._onboard_end_doc = onboard_end_doc
959 self._data_dir = data_dir
960 if data_dir and not os.path.isdir(data_dir):
961 os.mkdir(data_dir)
962
963 self._caching_model_wrapper_kw = dict(
964 cache_dir=self._data_dir,
965 strict_id_validation=strict_cache_id_validation,
966 id_hash_fn=caching.input_hash,
967 )
968
969 self._layouts = layouts if layouts else layout.DEFAULT_LAYOUTS
970 self._model_loaders: ModelLoadersMap = model_loaders or {}
971 self._models: dict[str, caching.CachingModelWrapper] = {}
972 for name, model in models.items():
973 if model_loaders is None:

Callers

nothing calls this directly

Calls 9

_run_annotatorsMethod · 0.95
_build_metadataMethod · 0.95
_validateMethod · 0.95
_warm_startMethod · 0.95
save_cacheMethod · 0.95
_warm_projectionsMethod · 0.95
make_handlerMethod · 0.95
infoMethod · 0.80
init_specMethod · 0.45

Tested by

no test coverage detected