Source: https://stackoverflow.com/a/54128391
| 24 | |
| 25 | |
| 26 | class NumpyExtension(Extension): |
| 27 | """Source: https://stackoverflow.com/a/54128391""" |
| 28 | |
| 29 | def __init__(self, *args, **kwargs): |
| 30 | self.__include_dirs = [] |
| 31 | super().__init__(*args, **kwargs) |
| 32 | |
| 33 | @property |
| 34 | def include_dirs(self): |
| 35 | import numpy |
| 36 | return self.__include_dirs + [numpy.get_include()] |
| 37 | |
| 38 | @include_dirs.setter |
| 39 | def include_dirs(self, dirs): |
| 40 | self.__include_dirs = dirs |
| 41 | |
| 42 | def config_fastbpe(): |
| 43 | try: |