"vscode:/vscode.git/clone" did not exist on "34e07701d3a383796097c5f10b601f4cbaa631a2"
trigger_import.py 254 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
"""Trigger import of some modules to write some caches,
so that static analysis (e.g., pyright) can know the type."""

import os
import sys

sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../../'))

import nni
import nni.retiarii.nn.pytorch