oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Type hint init_observations return type

Browse Source
This commit is contained in:
Marek Wolan
2023-06-01 09:57:33 +01:00
parent 72f76705a9
commit 5e506b1499
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/primaite/environment/primaite_env.py
View File

@@ -6,7 +6,7 @@ import csv
import logging import logging
import os.path import os.path
from datetime import datetime from datetime import datetime
from typing import Dict from typing import Dict, Tuple
import networkx as nx import networkx as nx
import numpy as np import numpy as np
@@ -641,7 +641,7 @@ class Primaite(Env):
else: else:
pass pass
def init_observations(self): def init_observations(self) -> Tuple[spaces.Space, np.ndarray]:
"""Build the observation space based on network laydown and provide initial obs. """Build the observation space based on network laydown and provide initial obs.
This method uses the object's `num_links`, `num_nodes`, `num_services`, This method uses the object's `num_links`, `num_nodes`, `num_services`,