ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 01:18:52 -06:00
Code Issues Packages Projects Releases Wiki Activity

Fix failing tests

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-08-05 14:05:50 -05:00
parent 475fe3d985
commit b6426462a1
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
miplearn/features/sample.py
View File

@@ -312,6 +312,7 @@ class Hdf5Sample(Sample):
def _pad(veclist: VectorList) -> Tuple[VectorList, List[int]]: def _pad(veclist: VectorList) -> Tuple[VectorList, List[int]]:
veclist = deepcopy(veclist)
lens = [len(v) if v is not None else -1 for v in veclist] lens = [len(v) if v is not None else -1 for v in veclist]
maxlen = max(lens) maxlen = max(lens)

2
tests/features/test_extractor.py
View File

@@ -192,7 +192,7 @@ if __name__ == "__main__":
extractor = FeaturesExtractor(with_lhs=False) extractor = FeaturesExtractor(with_lhs=False)
sample = Hdf5Sample("tmp/prof.h5", mode="w") sample = Hdf5Sample("tmp/prof.h5", mode="w")
def run(): def run() -> None:
extractor.extract_after_load_features(instance, solver, sample) extractor.extract_after_load_features(instance, solver, sample)
extractor.extract_after_lp_features(solver, sample) extractor.extract_after_lp_features(solver, sample)