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

Make get_instance_features return np.ndarray

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-08-09 14:02:14 -05:00
parent 47d3011808
commit 56b39b6c9c
6 changed files with 29 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
miplearn/instance/file.py
View File

@@ -6,6 +6,7 @@ import os
from typing import Any, Optional, List, Dict, TYPE_CHECKING
import pickle
import numpy as np
from overrides import overrides
from miplearn.features.sample import Hdf5Sample, Sample
@@ -30,7 +31,7 @@ class FileInstance(Instance):
return self.instance.to_model()
@overrides
def get_instance_features(self) -> List[float]:
def get_instance_features(self) -> np.ndarray:
assert self.instance is not None
return self.instance.get_instance_features()