Skip to content

Commit e831717

Browse files
committed
fix doc problems
Signed-off-by: reiase <[email protected]>
1 parent f3a6fb2 commit e831717

File tree

3 files changed

+21
-34
lines changed

3 files changed

+21
-34
lines changed

README_CN.md

Lines changed: 0 additions & 19 deletions
This file was deleted.

hyperparameter/hp.py

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,17 @@
11
import inspect
22
import json
33
import threading
4-
import inspect
54

6-
from typing import Any
5+
from typing import Any, Dict
6+
from typing import Callable
77

88

99
class Tracker:
10+
"""
11+
tracker for python read/write operations
12+
"""
1013
rlist = set()
14+
1115
wlist = set()
1216
callback = None
1317

@@ -38,9 +42,12 @@ def report():
3842
return '\n'.join(retvals)
3943

4044
@staticmethod
41-
def set_tracker(func):
45+
def set_tracker(func: Callable[[Dict[str, Any]], None]):
46+
""" report hyper-parameter value to a tracker, for example, `mlflow.tracking`
47+
"""
4248
Tracker.callback = func
4349

50+
4451
class Accessor(dict):
4552
"""
4653
Helper for accessing undefined parameters.
@@ -100,6 +107,7 @@ def __setattr__(self, name: str, value: Any):
100107
root[path] = HyperParameter()
101108
root = root[path]
102109
root[name] = value
110+
return value
103111

104112
def __str__(self):
105113
return ''
@@ -154,7 +162,8 @@ class HyperParameter(dict):
154162
'''
155163

156164
def __init__(self, **kws):
157-
return self.update(kws)
165+
super(HyperParameter, self).__init__()
166+
self.update(kws)
158167

159168
def update(self, kws):
160169
for k, v in kws.items():
@@ -220,9 +229,6 @@ def get(self, name: str) -> Any:
220229
Tracker.rlist.add(name)
221230
return obj[path[-1]] if path[-1] in obj else None
222231

223-
def holder(self):
224-
return
225-
226232
def __setitem__(self, key, value):
227233
if isinstance(value, dict):
228234
return dict.__setitem__(self, key, HyperParameter(**value))
@@ -247,11 +253,11 @@ def __setattr__(self, name, value):
247253

248254
def __call__(self) -> Any:
249255
"""
250-
Return a parameter accessor.
256+
Return a parameter accessor.
251257
252258
Returns:
253259
Any: holder of current parameter
254-
260+
255261
Examples:
256262
>>> cfg = HyperParameter(a=1, b = {'c':2, 'd': 3})
257263
>>> cfg().a.getOrElse(2)
@@ -279,9 +285,9 @@ def from_json_file(path):
279285

280286

281287
class param_scope(HyperParameter):
282-
'''
288+
'''
283289
thread safe scoped hyper parameeter
284-
290+
285291
Examples:
286292
create a scoped HyperParameter
287293
>>> with param_scope(**{'a': 1, 'b': 2}) as cfg:
@@ -492,4 +498,4 @@ def read_a():
492498
self.assertEqual(read_a(), 2)
493499
self.assertEqual(read_a(), 1)
494500

495-
unittest.main()
501+
unittest.main()

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,14 @@
11
from setuptools import setup, find_packages
22
from setuptools.command.install import install
33

4-
54
with open("README.md", "r", encoding="utf-8") as fh:
65
long_description = fh.read()
76

87
setup(
98
name="hyperparameter",
10-
version="0.1.4",
11-
description="A hyper-parameter library for researchers, data scientists and machine learning engineers.",
9+
version="0.2.0",
10+
description=
11+
"A hyper-parameter library for researchers, data scientists and machine learning engineers.",
1212
long_description=long_description,
1313
long_description_content_type="text/markdown",
1414
author="Reiase",

0 commit comments

Comments
 (0)