We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 37ce960 commit aad1bcaCopy full SHA for aad1bca
.github/workflows/export_constants.py
@@ -1,13 +1,18 @@
1
import json
2
3
-import chebai.preprocessing.reader as dr
+from chebai.preprocessing.reader import (
4
+ CLS_TOKEN,
5
+ EMBEDDING_OFFSET,
6
+ MASK_TOKEN_INDEX,
7
+ PADDING_TOKEN_INDEX,
8
+)
9
10
# Define the constants you want to export
11
constants = {
- "EMBEDDING_OFFSET": dr.EMBEDDING_OFFSET,
- "CLS_TOKEN": dr.CLS_TOKEN,
- "PADDING_TOKEN_INDEX": dr.PADDING_TOKEN_INDEX,
- "MASK_TOKEN_INDEX": dr.MASK_TOKEN_INDEX,
12
+ "EMBEDDING_OFFSET": EMBEDDING_OFFSET,
13
+ "CLS_TOKEN": CLS_TOKEN,
14
+ "PADDING_TOKEN_INDEX": PADDING_TOKEN_INDEX,
15
+ "MASK_TOKEN_INDEX": MASK_TOKEN_INDEX,
16
}
17
18
# Write constants to a JSON file
0 commit comments