浏览代码

Fix test typing

/develop/add-fire/policy-tests
Ervin Teng 4 年前
当前提交
554ca0b9
共有 1 个文件被更改,包括 1 次插入2 次删除
  1. 3
      ml-agents/mlagents/trainers/tests/torch/test_policy.py

3
ml-agents/mlagents/trainers/tests/torch/test_policy.py


import pytest
from typing import Tuple
import torch
from mlagents.trainers.policy.torch_policy import TorchPolicy

use_discrete: bool = True,
use_visual: bool = False,
seed: int = 0,
) -> Tuple[TorchPolicy, TrainerSettings]:
) -> TorchPolicy:
mock_spec = mb.setup_test_behavior_specs(
use_discrete,
use_visual,

正在加载...
取消
保存