(value)
| 43 | ########################################################################### |
| 44 | |
| 45 | def detect_framework(value): |
| 46 | classes = (value.__class__,) + value.__class__.__bases__ |
| 47 | for c in classes: |
| 48 | if c.__module__.startswith("torch"): |
| 49 | return "torch" |
| 50 | elif c.__module__.startswith("tensorflow"): |
| 51 | return "tensorflow" |
| 52 | |
| 53 | |
| 54 | ########################################################################### |
no outgoing calls
no test coverage detected