Skip to content

Feat/pr metrics#54

Merged
NotBioWaste905 merged 70 commits intodevfrom
feat/pr_metrics
May 5, 2025
Merged

Feat/pr metrics#54
NotBioWaste905 merged 70 commits intodevfrom
feat/pr_metrics

Conversation

@yupesh
Copy link
Copy Markdown
Collaborator

@yupesh yupesh commented Apr 24, 2025

No description provided.

from dialogue2graph.pipelines.helpers.parse_data import PipelineRawDataType
from dialogue2graph.pipelines.model_storage import ModelStorage

engine = create_engine(os.getenv("SQLALCHEMY_DATABASE_URI"))
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

wrap this code in try...except and use InMemoryCache if sql connection fails

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

done

from dialogue2graph.pipelines.core.algorithms import GraphGenerator


class DGBaseGenerator(GraphGenerator):
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

add docstring

pyproject.toml Outdated
# optimum-intel = {git = "https://github.com/huggingface/optimum-intel.git", extras = ["neural-compressor"]}
# torch = {version = "^2.7.0+cpu", source = "pytorch-cpu"}
# torchvision = {version = "^0.22.0+cpu", source = "pytorch-cpu"}
# torchaudio = {version = "^2.7.0+cpu", source = "pytorch-cpu"}
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

remove commented lines

# model = MockChatModel()
# generator = LoopedGraphGenerator(generation_model=model, validation_model=model, theme_validation_model=model)
# assert isinstance(generator, LoopedGraphGenerator)
def test_looped_graph_generator_init():
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Remove LoopedGraphGenerator test from here
Move it to a separate test file like test_dialogue_augmentation.py

@NotBioWaste905 NotBioWaste905 merged commit f7d6067 into dev May 5, 2025
15 of 16 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants