Get versions of main dependencies of BERTopic.
()
| 467 | |
| 468 | |
| 469 | def get_package_versions(): |
| 470 | """Get versions of main dependencies of BERTopic.""" |
| 471 | try: |
| 472 | import platform |
| 473 | from numpy import __version__ as np_version |
| 474 | from pandas import __version__ as pandas_version |
| 475 | from sklearn import __version__ as sklearn_version |
| 476 | from plotly import __version__ as plotly_version |
| 477 | |
| 478 | try: |
| 479 | from importlib.metadata import version |
| 480 | |
| 481 | hdbscan_version = version("hdbscan") |
| 482 | except (ImportError, ModuleNotFoundError): |
| 483 | hdbscan_version = None |
| 484 | |
| 485 | try: |
| 486 | from umap import __version__ as umap_version |
| 487 | except (ImportError, ModuleNotFoundError): |
| 488 | umap_version = None |
| 489 | |
| 490 | try: |
| 491 | from sentence_transformers import __version__ as sbert_version |
| 492 | except (ImportError, ModuleNotFoundError): |
| 493 | sbert_version = None |
| 494 | |
| 495 | try: |
| 496 | from numba import __version__ as numba_version |
| 497 | except (ImportError, ModuleNotFoundError): |
| 498 | numba_version = None |
| 499 | |
| 500 | try: |
| 501 | from transformers import __version__ as transformers_version |
| 502 | except (ImportError, ModuleNotFoundError): |
| 503 | transformers_version = None |
| 504 | |
| 505 | return { |
| 506 | "Numpy": np_version, |
| 507 | "HDBSCAN": hdbscan_version, |
| 508 | "UMAP": umap_version, |
| 509 | "Pandas": pandas_version, |
| 510 | "Scikit-Learn": sklearn_version, |
| 511 | "Sentence-transformers": sbert_version, |
| 512 | "Transformers": transformers_version, |
| 513 | "Numba": numba_version, |
| 514 | "Plotly": plotly_version, |
| 515 | "Python": platform.python_version(), |
| 516 | } |
| 517 | except Exception as e: |
| 518 | return e |
| 519 | |
| 520 | |
| 521 | def load_safetensors(path): |