Skip to content

Commit fc0ffdf

Browse files
committed
add verbose flag
1 parent 941432b commit fc0ffdf

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/pygraphon/estimators/networkhistogram/HistogramEstimator.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,10 @@ class HistogramEstimator(BaseEstimator):
2222
"""
2323

2424
def __init__(
25-
self, bandwithHist: Optional[float] = None, method: str = "mine"
25+
self, bandwithHist: Optional[float] = None, method: str = "mine", progress_bar: bool = False
2626
) -> None:
2727
super().__init__()
28+
self.progress_bar = progress_bar
2829
self.bandwidthHist = bandwithHist
2930
if method != "mine":
3031
DeprecationWarning("method other than 'mine' are deprecated")
@@ -111,6 +112,7 @@ def _approximate(
111112
absTol=absTol,
112113
maxNumIterations=maxNumIterations,
113114
past_non_improving=past_non_improving,
115+
progress_bar=self.progress_bar,
114116
)
115117

116118
bandwidthHist = h / adjacencyMatrix.shape[0]

0 commit comments

Comments
 (0)