generate-wandb-python-reference/configuration.py at main · ngrayluna/generate-wandb-python-reference · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
from pathlib import Path

BASE_DIR = Path(__file__).resolve().parents[1]  # go up from `generate-wandb-python-reference/config/config.py`

SOURCE = {
    "SDK": {
        "module": "wandb",
        "file_path": BASE_DIR / "wandb" / "wandb" / "__init__.template.pyi",
        "hugo_specs": {
            "title": "Actions",
            "description": "Use during training to log experiments, track metrics, and save model artifacts.",
            "frontmatter": "namespace: python_sdk_actions",
            "folder_name": "global",
        },
    },
    "CUSTOMCHARTS": {
        "module": "wandb.plot",
        "file_path": BASE_DIR / "wandb" / "wandb" / "plot" / "__init__.py",
        "hugo_specs": {
            "title": "Custom Charts",
            "description": "Create custom charts and visualizations.",
            "frontmatter": "namespace: python_sdk_custom_charts",
            "folder_name": "custom-charts",
        },
    },
    "DATATYPE": {
        "module": "wandb.sdk.data_types",
        "file_path": BASE_DIR / "wandb" / "wandb" / "__init__.template.pyi",
        "hugo_specs": {
            "title": "Data Types",
            "description": "Defines Data Types for logging interactive visualizations to W&B.",
            "frontmatter": "namespace: python_sdk_data_type",
            "folder_name": "data-types",
        },
    },
    "PUBLIC_API": {
        "module": "wandb.apis.public",
        "file_path": BASE_DIR / "wandb" / "wandb" / "apis" / "public" / "__init__.py",
        "hugo_specs": {
            "title": "Query API",
            "description": "Query and analyze data logged to W&B.",
            "frontmatter": "namespace: public_apis_namespace",
            "folder_name": "public-api",
        },
    },
    "AUTOMATIONS": {
        "module": "wandb.automations",
        "file_path": BASE_DIR / "wandb" / "wandb" / "automations" / "__init__.py",
        "hugo_specs": {
            "title": "Automations",
            "description": "Automate your W&B workflows.",
            "frontmatter": "namespace: automations_namespace",
            "folder_name": "automations",
        },
    },
}