diff --git a/src/databricks/labs/ucx/source_code/known.json b/src/databricks/labs/ucx/source_code/known.json index a6bbfdc7b0..57e8ffcd54 100644 --- a/src/databricks/labs/ucx/source_code/known.json +++ b/src/databricks/labs/ucx/source_code/known.json @@ -22509,6 +22509,92 @@ "safetensors.tensorflow": [], "safetensors.torch": [] }, + "salesforce-uni2ts": { + "uni2ts": [], + "uni2ts.common": [], + "uni2ts.common.core": [], + "uni2ts.common.env": [], + "uni2ts.common.hydra_util": [], + "uni2ts.common.sampler": [], + "uni2ts.common.torch_util": [], + "uni2ts.common.typing": [], + "uni2ts.data": [], + "uni2ts.data.builder": [], + "uni2ts.data.builder._base": [], + "uni2ts.data.builder.lotsa_v1": [], + "uni2ts.data.builder.lotsa_v1._base": [], + "uni2ts.data.builder.lotsa_v1.buildings_bench": [], + "uni2ts.data.builder.lotsa_v1.cloudops_tsf": [], + "uni2ts.data.builder.lotsa_v1.cmip6": [], + "uni2ts.data.builder.lotsa_v1.era5": [], + "uni2ts.data.builder.lotsa_v1.gluonts": [], + "uni2ts.data.builder.lotsa_v1.largest": [], + "uni2ts.data.builder.lotsa_v1.lib_city": [], + "uni2ts.data.builder.lotsa_v1.others": [], + "uni2ts.data.builder.lotsa_v1.proenfo": [], + "uni2ts.data.builder.lotsa_v1.subseasonal": [], + "uni2ts.data.builder.simple": [], + "uni2ts.data.dataset": [], + "uni2ts.data.indexer": [], + "uni2ts.data.indexer._base": [], + "uni2ts.data.indexer.hf_dataset_indexer": [], + "uni2ts.data.loader": [], + "uni2ts.distribution": [], + "uni2ts.distribution._base": [], + "uni2ts.distribution.laplace": [], + "uni2ts.distribution.log_normal": [], + "uni2ts.distribution.mixture": [], + "uni2ts.distribution.negative_binomial": [], + "uni2ts.distribution.normal": [], + "uni2ts.distribution.pareto": [], + "uni2ts.distribution.student_t": [], + "uni2ts.eval_util": [], + "uni2ts.eval_util._hf_dataset": [], + "uni2ts.eval_util._lsf_dataset": [], + "uni2ts.eval_util._pf_dataset": [], + "uni2ts.eval_util.data": [], + "uni2ts.eval_util.evaluation": [], + "uni2ts.eval_util.metrics": [], + "uni2ts.eval_util.plot": [], + "uni2ts.loss": [], + "uni2ts.loss.packed": [], + "uni2ts.loss.packed._base": [], + "uni2ts.loss.packed.distribution": [], + "uni2ts.loss.packed.normalized": [], + "uni2ts.loss.packed.percentage_error": [], + "uni2ts.loss.packed.point": [], + "uni2ts.model": [], + "uni2ts.model.moirai": [], + "uni2ts.model.moirai.finetune": [], + "uni2ts.model.moirai.forecast": [], + "uni2ts.model.moirai.module": [], + "uni2ts.model.moirai.pretrain": [], + "uni2ts.module": [], + "uni2ts.module.attention": [], + "uni2ts.module.ffn": [], + "uni2ts.module.norm": [], + "uni2ts.module.packed_scaler": [], + "uni2ts.module.position": [], + "uni2ts.module.position.additive": [], + "uni2ts.module.position.attn_bias": [], + "uni2ts.module.position.attn_projection": [], + "uni2ts.module.transformer": [], + "uni2ts.module.ts_embed": [], + "uni2ts.optim": [], + "uni2ts.optim.lr_scheduler": [], + "uni2ts.transform": [], + "uni2ts.transform._base": [], + "uni2ts.transform._mixin": [], + "uni2ts.transform.crop": [], + "uni2ts.transform.feature": [], + "uni2ts.transform.field": [], + "uni2ts.transform.imputation": [], + "uni2ts.transform.pad": [], + "uni2ts.transform.patch": [], + "uni2ts.transform.resample": [], + "uni2ts.transform.reshape": [], + "uni2ts.transform.task": [] + }, "scikit-learn": { "sklearn": [] },