wassname/optuna-dashboard
1
0
Fork 0
mirror of https://github.com/wassname/optuna-dashboard.git synced 2026-09-12 12:40:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #220 from himkt/mypy-typing

Browse Source 
Add annotation for distributions
This commit is contained in:
Masashi Shibata
2022-04-24 16:52:27 +09:00
committed by GitHub
parent ca19674bbb a90e1a0def
commit ac4c8cf551
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_tests/test_search_space.py
+6 -3
View File
@@ -1,8 +1,11 @@
from typing import Dict
from typing import List
from unittest import TestCase
import warnings
import optuna
from optuna import create_trial
from optuna.distributions import BaseDistribution
from optuna.distributions import UniformDistribution
from optuna.exceptions import ExperimentalWarning
from optuna.trial import TrialState
@@ -16,7 +19,7 @@ class SearchSpaceTestCase(TestCase):
warnings.simplefilter("ignore", category=ExperimentalWarning)
def test_same_distributions(self) -> None:
distributions = [
distributions: List[Dict[str, BaseDistribution]] = [
{
"x0": UniformDistribution(low=0, high=10),
"x1": UniformDistribution(low=0, high=10),
@@ -47,7 +50,7 @@ class SearchSpaceTestCase(TestCase):
self.assertEqual(len(search_space.union), 2)
def test_different_distributions(self) -> None:
distributions = [
distributions: List[Dict[str, BaseDistribution]] = [
{
"x0": UniformDistribution(low=0, high=10),
"x1": UniformDistribution(low=0, high=10),
@@ -78,7 +81,7 @@ class SearchSpaceTestCase(TestCase):
self.assertEqual(len(search_space.union), 3)
def test_dynamic_search_space(self) -> None:
distributions = [
distributions: List[Dict[str, BaseDistribution]] = [
{
"x0": UniformDistribution(low=0, high=10),
"x1": UniformDistribution(low=0, high=10),