Skip to content
This repository was archived by the owner on Aug 25, 2024. It is now read-only.
This repository was archived by the owner on Aug 25, 2024. It is now read-only.

model: scikit: setup.py needs correct entrypoints #165

@johnandersen777

Description

@johnandersen777
diff --git a/model/scikit/setup.py b/model/scikit/setup.py
index 1bd6284..efbff80 100644
--- a/model/scikit/setup.py
+++ b/model/scikit/setup.py
@@ -57,5 +57,9 @@ setup(
     ],
     install_requires=INSTALL_REQUIRES,
     packages=find_packages(),
-    entry_points={"dffml.model": [f"scikitlr = {IMPORT_NAME}.sciLR:LR"]},
+    entry_points={"dffml.model": [
+        f"scikitmlp = {IMPORT_NAME}.scikit_models:MLPClassifierModel",
+        f"scikitmlp = {IMPORT_NAME}.scikit_models:MLPClassifierModel",
+        f"scikitmlp = {IMPORT_NAME}.scikit_models:MLPClassifierModel",
+        ]},
 )

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions