relationalai 0.12.13__py3-none-any.whl → 0.13.0.dev0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- relationalai/__init__.py +1 -209
- relationalai/config/__init__.py +56 -0
- relationalai/config/config.py +289 -0
- relationalai/config/config_fields.py +86 -0
- relationalai/config/connections/__init__.py +46 -0
- relationalai/config/connections/base.py +23 -0
- relationalai/config/connections/duckdb.py +29 -0
- relationalai/config/connections/snowflake.py +243 -0
- relationalai/config/external/__init__.py +17 -0
- relationalai/config/external/dbt_converter.py +101 -0
- relationalai/config/external/dbt_models.py +93 -0
- relationalai/config/external/snowflake_converter.py +41 -0
- relationalai/config/external/snowflake_models.py +85 -0
- relationalai/config/external/utils.py +19 -0
- relationalai/semantics/__init__.py +146 -22
- relationalai/semantics/backends/lqp/annotations.py +11 -0
- relationalai/semantics/backends/sql/sql_compiler.py +327 -0
- relationalai/semantics/frontend/base.py +1707 -0
- relationalai/semantics/frontend/core.py +179 -0
- relationalai/semantics/frontend/front_compiler.py +1313 -0
- relationalai/semantics/frontend/pprint.py +408 -0
- relationalai/semantics/metamodel/__init__.py +6 -40
- relationalai/semantics/metamodel/builtins.py +205 -769
- relationalai/semantics/metamodel/metamodel.py +437 -0
- relationalai/semantics/metamodel/metamodel_analyzer.py +519 -0
- relationalai/semantics/metamodel/pprint.py +412 -0
- relationalai/semantics/metamodel/rewriter.py +266 -0
- relationalai/semantics/metamodel/typer.py +1378 -0
- relationalai/semantics/std/__init__.py +60 -40
- relationalai/semantics/std/aggregates.py +149 -0
- relationalai/semantics/std/common.py +44 -0
- relationalai/semantics/std/constraints.py +37 -43
- relationalai/semantics/std/datetime.py +246 -135
- relationalai/semantics/std/decimals.py +45 -52
- relationalai/semantics/std/floats.py +13 -5
- relationalai/semantics/std/integers.py +26 -11
- relationalai/semantics/std/math.py +183 -112
- relationalai/semantics/std/numbers.py +86 -0
- relationalai/semantics/std/re.py +80 -62
- relationalai/semantics/std/strings.py +117 -60
- relationalai/shims/executor.py +147 -0
- relationalai/shims/helpers.py +126 -0
- relationalai/shims/hoister.py +221 -0
- relationalai/shims/mm2v0.py +1290 -0
- relationalai/tools/cli/__init__.py +6 -0
- relationalai/tools/cli/cli.py +90 -0
- relationalai/tools/cli/components/__init__.py +5 -0
- relationalai/tools/cli/components/progress_reader.py +1524 -0
- relationalai/tools/cli/components/utils.py +58 -0
- relationalai/tools/cli/config_template.py +45 -0
- relationalai/tools/cli/dev.py +19 -0
- relationalai/tools/debugger.py +289 -183
- relationalai/tools/typer_debugger.py +93 -0
- relationalai/util/dataclasses.py +43 -0
- relationalai/util/docutils.py +40 -0
- relationalai/util/error.py +199 -0
- relationalai/util/format.py +48 -106
- relationalai/util/naming.py +145 -0
- relationalai/util/python.py +35 -0
- relationalai/util/runtime.py +156 -0
- relationalai/util/schema.py +197 -0
- relationalai/util/source.py +185 -0
- relationalai/util/structures.py +163 -0
- relationalai/util/tracing.py +261 -0
- relationalai-0.13.0.dev0.dist-info/METADATA +46 -0
- relationalai-0.13.0.dev0.dist-info/RECORD +488 -0
- relationalai-0.13.0.dev0.dist-info/WHEEL +5 -0
- relationalai-0.13.0.dev0.dist-info/entry_points.txt +3 -0
- relationalai-0.13.0.dev0.dist-info/top_level.txt +2 -0
- v0/relationalai/__init__.py +216 -0
- v0/relationalai/clients/azure.py +477 -0
- v0/relationalai/clients/client.py +912 -0
- v0/relationalai/clients/config.py +673 -0
- v0/relationalai/clients/direct_access_client.py +118 -0
- v0/relationalai/clients/hash_util.py +31 -0
- v0/relationalai/clients/local.py +571 -0
- v0/relationalai/clients/profile_polling.py +73 -0
- v0/relationalai/clients/result_helpers.py +420 -0
- v0/relationalai/clients/snowflake.py +3869 -0
- v0/relationalai/clients/types.py +113 -0
- v0/relationalai/clients/use_index_poller.py +980 -0
- v0/relationalai/clients/util.py +356 -0
- v0/relationalai/debugging.py +389 -0
- v0/relationalai/dsl.py +1749 -0
- v0/relationalai/early_access/builder/__init__.py +30 -0
- v0/relationalai/early_access/builder/builder/__init__.py +35 -0
- v0/relationalai/early_access/builder/snowflake/__init__.py +12 -0
- v0/relationalai/early_access/builder/std/__init__.py +25 -0
- v0/relationalai/early_access/builder/std/decimals/__init__.py +12 -0
- v0/relationalai/early_access/builder/std/integers/__init__.py +12 -0
- v0/relationalai/early_access/builder/std/math/__init__.py +12 -0
- v0/relationalai/early_access/builder/std/strings/__init__.py +14 -0
- v0/relationalai/early_access/devtools/__init__.py +12 -0
- v0/relationalai/early_access/devtools/benchmark_lqp/__init__.py +12 -0
- v0/relationalai/early_access/devtools/extract_lqp/__init__.py +12 -0
- v0/relationalai/early_access/dsl/adapters/orm/adapter_qb.py +427 -0
- v0/relationalai/early_access/dsl/adapters/orm/parser.py +636 -0
- v0/relationalai/early_access/dsl/adapters/owl/adapter.py +176 -0
- v0/relationalai/early_access/dsl/adapters/owl/parser.py +160 -0
- v0/relationalai/early_access/dsl/bindings/common.py +402 -0
- v0/relationalai/early_access/dsl/bindings/csv.py +170 -0
- v0/relationalai/early_access/dsl/bindings/legacy/binding_models.py +143 -0
- v0/relationalai/early_access/dsl/bindings/snowflake.py +64 -0
- v0/relationalai/early_access/dsl/codegen/binder.py +411 -0
- v0/relationalai/early_access/dsl/codegen/common.py +79 -0
- v0/relationalai/early_access/dsl/codegen/helpers.py +23 -0
- v0/relationalai/early_access/dsl/codegen/relations.py +700 -0
- v0/relationalai/early_access/dsl/codegen/weaver.py +417 -0
- v0/relationalai/early_access/dsl/core/builders/__init__.py +47 -0
- v0/relationalai/early_access/dsl/core/builders/logic.py +19 -0
- v0/relationalai/early_access/dsl/core/builders/scalar_constraint.py +11 -0
- v0/relationalai/early_access/dsl/core/constraints/predicate/atomic.py +455 -0
- v0/relationalai/early_access/dsl/core/constraints/predicate/universal.py +73 -0
- v0/relationalai/early_access/dsl/core/constraints/scalar.py +310 -0
- v0/relationalai/early_access/dsl/core/context.py +13 -0
- v0/relationalai/early_access/dsl/core/cset.py +132 -0
- v0/relationalai/early_access/dsl/core/exprs/__init__.py +116 -0
- v0/relationalai/early_access/dsl/core/exprs/relational.py +18 -0
- v0/relationalai/early_access/dsl/core/exprs/scalar.py +412 -0
- v0/relationalai/early_access/dsl/core/instances.py +44 -0
- v0/relationalai/early_access/dsl/core/logic/__init__.py +193 -0
- v0/relationalai/early_access/dsl/core/logic/aggregation.py +98 -0
- v0/relationalai/early_access/dsl/core/logic/exists.py +223 -0
- v0/relationalai/early_access/dsl/core/logic/helper.py +163 -0
- v0/relationalai/early_access/dsl/core/namespaces.py +32 -0
- v0/relationalai/early_access/dsl/core/relations.py +276 -0
- v0/relationalai/early_access/dsl/core/rules.py +112 -0
- v0/relationalai/early_access/dsl/core/std/__init__.py +45 -0
- v0/relationalai/early_access/dsl/core/temporal/recall.py +6 -0
- v0/relationalai/early_access/dsl/core/types/__init__.py +270 -0
- v0/relationalai/early_access/dsl/core/types/concepts.py +128 -0
- v0/relationalai/early_access/dsl/core/types/constrained/__init__.py +267 -0
- v0/relationalai/early_access/dsl/core/types/constrained/nominal.py +143 -0
- v0/relationalai/early_access/dsl/core/types/constrained/subtype.py +124 -0
- v0/relationalai/early_access/dsl/core/types/standard.py +92 -0
- v0/relationalai/early_access/dsl/core/types/unconstrained.py +50 -0
- v0/relationalai/early_access/dsl/core/types/variables.py +203 -0
- v0/relationalai/early_access/dsl/ir/compiler.py +318 -0
- v0/relationalai/early_access/dsl/ir/executor.py +260 -0
- v0/relationalai/early_access/dsl/ontologies/constraints.py +88 -0
- v0/relationalai/early_access/dsl/ontologies/export.py +30 -0
- v0/relationalai/early_access/dsl/ontologies/models.py +453 -0
- v0/relationalai/early_access/dsl/ontologies/python_printer.py +303 -0
- v0/relationalai/early_access/dsl/ontologies/readings.py +60 -0
- v0/relationalai/early_access/dsl/ontologies/relationships.py +322 -0
- v0/relationalai/early_access/dsl/ontologies/roles.py +87 -0
- v0/relationalai/early_access/dsl/ontologies/subtyping.py +55 -0
- v0/relationalai/early_access/dsl/orm/constraints.py +438 -0
- v0/relationalai/early_access/dsl/orm/measures/dimensions.py +200 -0
- v0/relationalai/early_access/dsl/orm/measures/initializer.py +16 -0
- v0/relationalai/early_access/dsl/orm/measures/measure_rules.py +275 -0
- v0/relationalai/early_access/dsl/orm/measures/measures.py +299 -0
- v0/relationalai/early_access/dsl/orm/measures/role_exprs.py +268 -0
- v0/relationalai/early_access/dsl/orm/models.py +256 -0
- v0/relationalai/early_access/dsl/orm/object_oriented_printer.py +344 -0
- v0/relationalai/early_access/dsl/orm/printer.py +469 -0
- v0/relationalai/early_access/dsl/orm/reasoners.py +480 -0
- v0/relationalai/early_access/dsl/orm/relations.py +19 -0
- v0/relationalai/early_access/dsl/orm/relationships.py +251 -0
- v0/relationalai/early_access/dsl/orm/types.py +42 -0
- v0/relationalai/early_access/dsl/orm/utils.py +79 -0
- v0/relationalai/early_access/dsl/orm/verb.py +204 -0
- v0/relationalai/early_access/dsl/physical_metadata/tables.py +133 -0
- v0/relationalai/early_access/dsl/relations.py +170 -0
- v0/relationalai/early_access/dsl/rulesets.py +69 -0
- v0/relationalai/early_access/dsl/schemas/__init__.py +450 -0
- v0/relationalai/early_access/dsl/schemas/builder.py +48 -0
- v0/relationalai/early_access/dsl/schemas/comp_names.py +51 -0
- v0/relationalai/early_access/dsl/schemas/components.py +203 -0
- v0/relationalai/early_access/dsl/schemas/contexts.py +156 -0
- v0/relationalai/early_access/dsl/schemas/exprs.py +89 -0
- v0/relationalai/early_access/dsl/schemas/fragments.py +464 -0
- v0/relationalai/early_access/dsl/serialization.py +79 -0
- v0/relationalai/early_access/dsl/serialize/exporter.py +163 -0
- v0/relationalai/early_access/dsl/snow/api.py +104 -0
- v0/relationalai/early_access/dsl/snow/common.py +76 -0
- v0/relationalai/early_access/dsl/state_mgmt/__init__.py +129 -0
- v0/relationalai/early_access/dsl/state_mgmt/state_charts.py +125 -0
- v0/relationalai/early_access/dsl/state_mgmt/transitions.py +130 -0
- v0/relationalai/early_access/dsl/types/__init__.py +40 -0
- v0/relationalai/early_access/dsl/types/concepts.py +12 -0
- v0/relationalai/early_access/dsl/types/entities.py +135 -0
- v0/relationalai/early_access/dsl/types/values.py +17 -0
- v0/relationalai/early_access/dsl/utils.py +102 -0
- v0/relationalai/early_access/graphs/__init__.py +13 -0
- v0/relationalai/early_access/lqp/__init__.py +12 -0
- v0/relationalai/early_access/lqp/compiler/__init__.py +12 -0
- v0/relationalai/early_access/lqp/constructors/__init__.py +18 -0
- v0/relationalai/early_access/lqp/executor/__init__.py +12 -0
- v0/relationalai/early_access/lqp/ir/__init__.py +12 -0
- v0/relationalai/early_access/lqp/passes/__init__.py +12 -0
- v0/relationalai/early_access/lqp/pragmas/__init__.py +12 -0
- v0/relationalai/early_access/lqp/primitives/__init__.py +12 -0
- v0/relationalai/early_access/lqp/types/__init__.py +12 -0
- v0/relationalai/early_access/lqp/utils/__init__.py +12 -0
- v0/relationalai/early_access/lqp/validators/__init__.py +12 -0
- v0/relationalai/early_access/metamodel/__init__.py +58 -0
- v0/relationalai/early_access/metamodel/builtins/__init__.py +12 -0
- v0/relationalai/early_access/metamodel/compiler/__init__.py +12 -0
- v0/relationalai/early_access/metamodel/dependency/__init__.py +12 -0
- v0/relationalai/early_access/metamodel/factory/__init__.py +17 -0
- v0/relationalai/early_access/metamodel/helpers/__init__.py +12 -0
- v0/relationalai/early_access/metamodel/ir/__init__.py +14 -0
- v0/relationalai/early_access/metamodel/rewrite/__init__.py +7 -0
- v0/relationalai/early_access/metamodel/typer/__init__.py +3 -0
- v0/relationalai/early_access/metamodel/typer/typer/__init__.py +12 -0
- v0/relationalai/early_access/metamodel/types/__init__.py +15 -0
- v0/relationalai/early_access/metamodel/util/__init__.py +15 -0
- v0/relationalai/early_access/metamodel/visitor/__init__.py +12 -0
- v0/relationalai/early_access/rel/__init__.py +12 -0
- v0/relationalai/early_access/rel/executor/__init__.py +12 -0
- v0/relationalai/early_access/rel/rel_utils/__init__.py +12 -0
- v0/relationalai/early_access/rel/rewrite/__init__.py +7 -0
- v0/relationalai/early_access/solvers/__init__.py +19 -0
- v0/relationalai/early_access/sql/__init__.py +11 -0
- v0/relationalai/early_access/sql/executor/__init__.py +3 -0
- v0/relationalai/early_access/sql/rewrite/__init__.py +3 -0
- v0/relationalai/early_access/tests/logging/__init__.py +12 -0
- v0/relationalai/early_access/tests/test_snapshot_base/__init__.py +12 -0
- v0/relationalai/early_access/tests/utils/__init__.py +12 -0
- v0/relationalai/environments/__init__.py +35 -0
- v0/relationalai/environments/base.py +381 -0
- v0/relationalai/environments/colab.py +14 -0
- v0/relationalai/environments/generic.py +71 -0
- v0/relationalai/environments/ipython.py +68 -0
- v0/relationalai/environments/jupyter.py +9 -0
- v0/relationalai/environments/snowbook.py +169 -0
- v0/relationalai/errors.py +2455 -0
- v0/relationalai/experimental/SF.py +38 -0
- v0/relationalai/experimental/inspect.py +47 -0
- v0/relationalai/experimental/pathfinder/__init__.py +158 -0
- v0/relationalai/experimental/pathfinder/api.py +160 -0
- v0/relationalai/experimental/pathfinder/automaton.py +584 -0
- v0/relationalai/experimental/pathfinder/bridge.py +226 -0
- v0/relationalai/experimental/pathfinder/compiler.py +416 -0
- v0/relationalai/experimental/pathfinder/datalog.py +214 -0
- v0/relationalai/experimental/pathfinder/diagnostics.py +56 -0
- v0/relationalai/experimental/pathfinder/filter.py +236 -0
- v0/relationalai/experimental/pathfinder/glushkov.py +439 -0
- v0/relationalai/experimental/pathfinder/options.py +265 -0
- v0/relationalai/experimental/pathfinder/rpq.py +344 -0
- v0/relationalai/experimental/pathfinder/transition.py +200 -0
- v0/relationalai/experimental/pathfinder/utils.py +26 -0
- v0/relationalai/experimental/paths/api.py +143 -0
- v0/relationalai/experimental/paths/benchmarks/grid_graph.py +37 -0
- v0/relationalai/experimental/paths/examples/basic_example.py +40 -0
- v0/relationalai/experimental/paths/examples/minimal_engine_warmup.py +3 -0
- v0/relationalai/experimental/paths/examples/movie_example.py +77 -0
- v0/relationalai/experimental/paths/examples/paths_benchmark.py +115 -0
- v0/relationalai/experimental/paths/examples/paths_example.py +116 -0
- v0/relationalai/experimental/paths/examples/pattern_to_automaton.py +28 -0
- v0/relationalai/experimental/paths/find_paths_via_automaton.py +85 -0
- v0/relationalai/experimental/paths/graph.py +185 -0
- v0/relationalai/experimental/paths/path_algorithms/find_paths.py +280 -0
- v0/relationalai/experimental/paths/path_algorithms/one_sided_ball_repetition.py +26 -0
- v0/relationalai/experimental/paths/path_algorithms/one_sided_ball_upto.py +111 -0
- v0/relationalai/experimental/paths/path_algorithms/single.py +59 -0
- v0/relationalai/experimental/paths/path_algorithms/two_sided_balls_repetition.py +39 -0
- v0/relationalai/experimental/paths/path_algorithms/two_sided_balls_upto.py +103 -0
- v0/relationalai/experimental/paths/path_algorithms/usp-old.py +130 -0
- v0/relationalai/experimental/paths/path_algorithms/usp-tuple.py +183 -0
- v0/relationalai/experimental/paths/path_algorithms/usp.py +150 -0
- v0/relationalai/experimental/paths/product_graph.py +93 -0
- v0/relationalai/experimental/paths/rpq/automaton.py +584 -0
- v0/relationalai/experimental/paths/rpq/diagnostics.py +56 -0
- v0/relationalai/experimental/paths/rpq/rpq.py +378 -0
- v0/relationalai/experimental/paths/tests/tests_limit_sp_max_length.py +90 -0
- v0/relationalai/experimental/paths/tests/tests_limit_sp_multiple.py +119 -0
- v0/relationalai/experimental/paths/tests/tests_limit_sp_single.py +104 -0
- v0/relationalai/experimental/paths/tests/tests_limit_walks_multiple.py +113 -0
- v0/relationalai/experimental/paths/tests/tests_limit_walks_single.py +149 -0
- v0/relationalai/experimental/paths/tests/tests_one_sided_ball_repetition_multiple.py +70 -0
- v0/relationalai/experimental/paths/tests/tests_one_sided_ball_repetition_single.py +64 -0
- v0/relationalai/experimental/paths/tests/tests_one_sided_ball_upto_multiple.py +115 -0
- v0/relationalai/experimental/paths/tests/tests_one_sided_ball_upto_single.py +75 -0
- v0/relationalai/experimental/paths/tests/tests_single_paths.py +152 -0
- v0/relationalai/experimental/paths/tests/tests_single_walks.py +208 -0
- v0/relationalai/experimental/paths/tests/tests_single_walks_undirected.py +297 -0
- v0/relationalai/experimental/paths/tests/tests_two_sided_balls_repetition_multiple.py +107 -0
- v0/relationalai/experimental/paths/tests/tests_two_sided_balls_repetition_single.py +76 -0
- v0/relationalai/experimental/paths/tests/tests_two_sided_balls_upto_multiple.py +76 -0
- v0/relationalai/experimental/paths/tests/tests_two_sided_balls_upto_single.py +110 -0
- v0/relationalai/experimental/paths/tests/tests_usp_nsp_multiple.py +229 -0
- v0/relationalai/experimental/paths/tests/tests_usp_nsp_single.py +108 -0
- v0/relationalai/experimental/paths/tree_agg.py +168 -0
- v0/relationalai/experimental/paths/utilities/iterators.py +27 -0
- v0/relationalai/experimental/paths/utilities/prefix_sum.py +91 -0
- v0/relationalai/experimental/solvers.py +1087 -0
- v0/relationalai/loaders/__init__.py +0 -0
- v0/relationalai/loaders/csv.py +195 -0
- v0/relationalai/loaders/loader.py +177 -0
- v0/relationalai/loaders/types.py +23 -0
- v0/relationalai/rel_emitter.py +373 -0
- v0/relationalai/rel_utils.py +185 -0
- v0/relationalai/semantics/__init__.py +29 -0
- v0/relationalai/semantics/devtools/benchmark_lqp.py +536 -0
- v0/relationalai/semantics/devtools/compilation_manager.py +294 -0
- v0/relationalai/semantics/devtools/extract_lqp.py +110 -0
- v0/relationalai/semantics/internal/internal.py +3785 -0
- v0/relationalai/semantics/internal/snowflake.py +324 -0
- v0/relationalai/semantics/lqp/builtins.py +16 -0
- v0/relationalai/semantics/lqp/compiler.py +22 -0
- v0/relationalai/semantics/lqp/constructors.py +68 -0
- v0/relationalai/semantics/lqp/executor.py +469 -0
- v0/relationalai/semantics/lqp/intrinsics.py +24 -0
- v0/relationalai/semantics/lqp/model2lqp.py +839 -0
- v0/relationalai/semantics/lqp/passes.py +680 -0
- v0/relationalai/semantics/lqp/primitives.py +252 -0
- v0/relationalai/semantics/lqp/result_helpers.py +202 -0
- v0/relationalai/semantics/lqp/rewrite/annotate_constraints.py +57 -0
- v0/relationalai/semantics/lqp/rewrite/cdc.py +216 -0
- v0/relationalai/semantics/lqp/rewrite/extract_common.py +338 -0
- v0/relationalai/semantics/lqp/rewrite/extract_keys.py +449 -0
- v0/relationalai/semantics/lqp/rewrite/function_annotations.py +114 -0
- v0/relationalai/semantics/lqp/rewrite/functional_dependencies.py +314 -0
- v0/relationalai/semantics/lqp/rewrite/quantify_vars.py +296 -0
- v0/relationalai/semantics/lqp/rewrite/splinter.py +76 -0
- v0/relationalai/semantics/lqp/types.py +101 -0
- v0/relationalai/semantics/lqp/utils.py +160 -0
- v0/relationalai/semantics/lqp/validators.py +57 -0
- v0/relationalai/semantics/metamodel/__init__.py +40 -0
- v0/relationalai/semantics/metamodel/builtins.py +774 -0
- v0/relationalai/semantics/metamodel/compiler.py +133 -0
- v0/relationalai/semantics/metamodel/dependency.py +862 -0
- v0/relationalai/semantics/metamodel/executor.py +61 -0
- v0/relationalai/semantics/metamodel/factory.py +287 -0
- v0/relationalai/semantics/metamodel/helpers.py +361 -0
- v0/relationalai/semantics/metamodel/rewrite/discharge_constraints.py +39 -0
- v0/relationalai/semantics/metamodel/rewrite/dnf_union_splitter.py +210 -0
- v0/relationalai/semantics/metamodel/rewrite/extract_nested_logicals.py +78 -0
- v0/relationalai/semantics/metamodel/rewrite/flatten.py +549 -0
- v0/relationalai/semantics/metamodel/rewrite/format_outputs.py +165 -0
- v0/relationalai/semantics/metamodel/typer/checker.py +353 -0
- v0/relationalai/semantics/metamodel/typer/typer.py +1395 -0
- v0/relationalai/semantics/reasoners/__init__.py +10 -0
- v0/relationalai/semantics/reasoners/graph/__init__.py +37 -0
- v0/relationalai/semantics/reasoners/graph/core.py +9020 -0
- v0/relationalai/semantics/reasoners/optimization/__init__.py +68 -0
- v0/relationalai/semantics/reasoners/optimization/common.py +88 -0
- v0/relationalai/semantics/reasoners/optimization/solvers_dev.py +568 -0
- v0/relationalai/semantics/reasoners/optimization/solvers_pb.py +1163 -0
- v0/relationalai/semantics/rel/builtins.py +40 -0
- v0/relationalai/semantics/rel/compiler.py +989 -0
- v0/relationalai/semantics/rel/executor.py +359 -0
- v0/relationalai/semantics/rel/rel.py +482 -0
- v0/relationalai/semantics/rel/rel_utils.py +276 -0
- v0/relationalai/semantics/snowflake/__init__.py +3 -0
- v0/relationalai/semantics/sql/compiler.py +2503 -0
- v0/relationalai/semantics/sql/executor/duck_db.py +52 -0
- v0/relationalai/semantics/sql/executor/result_helpers.py +64 -0
- v0/relationalai/semantics/sql/executor/snowflake.py +145 -0
- v0/relationalai/semantics/sql/rewrite/denormalize.py +222 -0
- v0/relationalai/semantics/sql/rewrite/double_negation.py +49 -0
- v0/relationalai/semantics/sql/rewrite/recursive_union.py +127 -0
- v0/relationalai/semantics/sql/rewrite/sort_output_query.py +246 -0
- v0/relationalai/semantics/sql/sql.py +504 -0
- v0/relationalai/semantics/std/__init__.py +54 -0
- v0/relationalai/semantics/std/constraints.py +43 -0
- v0/relationalai/semantics/std/datetime.py +363 -0
- v0/relationalai/semantics/std/decimals.py +62 -0
- v0/relationalai/semantics/std/floats.py +7 -0
- v0/relationalai/semantics/std/integers.py +22 -0
- v0/relationalai/semantics/std/math.py +141 -0
- v0/relationalai/semantics/std/pragmas.py +11 -0
- v0/relationalai/semantics/std/re.py +83 -0
- v0/relationalai/semantics/std/std.py +14 -0
- v0/relationalai/semantics/std/strings.py +63 -0
- v0/relationalai/semantics/tests/__init__.py +0 -0
- v0/relationalai/semantics/tests/test_snapshot_abstract.py +143 -0
- v0/relationalai/semantics/tests/test_snapshot_base.py +9 -0
- v0/relationalai/semantics/tests/utils.py +46 -0
- v0/relationalai/std/__init__.py +70 -0
- v0/relationalai/tools/__init__.py +0 -0
- v0/relationalai/tools/cli.py +1940 -0
- v0/relationalai/tools/cli_controls.py +1826 -0
- v0/relationalai/tools/cli_helpers.py +390 -0
- v0/relationalai/tools/debugger.py +183 -0
- v0/relationalai/tools/debugger_client.py +109 -0
- v0/relationalai/tools/debugger_server.py +302 -0
- v0/relationalai/tools/dev.py +685 -0
- v0/relationalai/tools/qb_debugger.py +425 -0
- v0/relationalai/util/clean_up_databases.py +95 -0
- v0/relationalai/util/format.py +123 -0
- v0/relationalai/util/list_databases.py +9 -0
- v0/relationalai/util/otel_configuration.py +25 -0
- v0/relationalai/util/otel_handler.py +484 -0
- v0/relationalai/util/snowflake_handler.py +88 -0
- v0/relationalai/util/span_format_test.py +43 -0
- v0/relationalai/util/span_tracker.py +207 -0
- v0/relationalai/util/spans_file_handler.py +72 -0
- v0/relationalai/util/tracing_handler.py +34 -0
- frontend/debugger/dist/.gitignore +0 -2
- frontend/debugger/dist/assets/favicon-Dy0ZgA6N.png +0 -0
- frontend/debugger/dist/assets/index-Cssla-O7.js +0 -208
- frontend/debugger/dist/assets/index-DlHsYx1V.css +0 -9
- frontend/debugger/dist/index.html +0 -17
- relationalai/clients/azure.py +0 -477
- relationalai/clients/client.py +0 -912
- relationalai/clients/config.py +0 -673
- relationalai/clients/direct_access_client.py +0 -118
- relationalai/clients/export_procedure.py.jinja +0 -249
- relationalai/clients/hash_util.py +0 -31
- relationalai/clients/local.py +0 -571
- relationalai/clients/profile_polling.py +0 -73
- relationalai/clients/result_helpers.py +0 -420
- relationalai/clients/snowflake.py +0 -3869
- relationalai/clients/types.py +0 -113
- relationalai/clients/use_index_poller.py +0 -980
- relationalai/clients/util.py +0 -356
- relationalai/debugging.py +0 -389
- relationalai/dsl.py +0 -1749
- relationalai/early_access/builder/__init__.py +0 -30
- relationalai/early_access/builder/builder/__init__.py +0 -35
- relationalai/early_access/builder/snowflake/__init__.py +0 -12
- relationalai/early_access/builder/std/__init__.py +0 -25
- relationalai/early_access/builder/std/decimals/__init__.py +0 -12
- relationalai/early_access/builder/std/integers/__init__.py +0 -12
- relationalai/early_access/builder/std/math/__init__.py +0 -12
- relationalai/early_access/builder/std/strings/__init__.py +0 -14
- relationalai/early_access/devtools/__init__.py +0 -12
- relationalai/early_access/devtools/benchmark_lqp/__init__.py +0 -12
- relationalai/early_access/devtools/extract_lqp/__init__.py +0 -12
- relationalai/early_access/dsl/adapters/orm/adapter_qb.py +0 -427
- relationalai/early_access/dsl/adapters/orm/parser.py +0 -636
- relationalai/early_access/dsl/adapters/owl/adapter.py +0 -176
- relationalai/early_access/dsl/adapters/owl/parser.py +0 -160
- relationalai/early_access/dsl/bindings/common.py +0 -402
- relationalai/early_access/dsl/bindings/csv.py +0 -170
- relationalai/early_access/dsl/bindings/legacy/binding_models.py +0 -143
- relationalai/early_access/dsl/bindings/snowflake.py +0 -64
- relationalai/early_access/dsl/codegen/binder.py +0 -411
- relationalai/early_access/dsl/codegen/common.py +0 -79
- relationalai/early_access/dsl/codegen/helpers.py +0 -23
- relationalai/early_access/dsl/codegen/relations.py +0 -700
- relationalai/early_access/dsl/codegen/weaver.py +0 -417
- relationalai/early_access/dsl/core/builders/__init__.py +0 -47
- relationalai/early_access/dsl/core/builders/logic.py +0 -19
- relationalai/early_access/dsl/core/builders/scalar_constraint.py +0 -11
- relationalai/early_access/dsl/core/constraints/predicate/atomic.py +0 -455
- relationalai/early_access/dsl/core/constraints/predicate/universal.py +0 -73
- relationalai/early_access/dsl/core/constraints/scalar.py +0 -310
- relationalai/early_access/dsl/core/context.py +0 -13
- relationalai/early_access/dsl/core/cset.py +0 -132
- relationalai/early_access/dsl/core/exprs/__init__.py +0 -116
- relationalai/early_access/dsl/core/exprs/relational.py +0 -18
- relationalai/early_access/dsl/core/exprs/scalar.py +0 -412
- relationalai/early_access/dsl/core/instances.py +0 -44
- relationalai/early_access/dsl/core/logic/__init__.py +0 -193
- relationalai/early_access/dsl/core/logic/aggregation.py +0 -98
- relationalai/early_access/dsl/core/logic/exists.py +0 -223
- relationalai/early_access/dsl/core/logic/helper.py +0 -163
- relationalai/early_access/dsl/core/namespaces.py +0 -32
- relationalai/early_access/dsl/core/relations.py +0 -276
- relationalai/early_access/dsl/core/rules.py +0 -112
- relationalai/early_access/dsl/core/std/__init__.py +0 -45
- relationalai/early_access/dsl/core/temporal/recall.py +0 -6
- relationalai/early_access/dsl/core/types/__init__.py +0 -270
- relationalai/early_access/dsl/core/types/concepts.py +0 -128
- relationalai/early_access/dsl/core/types/constrained/__init__.py +0 -267
- relationalai/early_access/dsl/core/types/constrained/nominal.py +0 -143
- relationalai/early_access/dsl/core/types/constrained/subtype.py +0 -124
- relationalai/early_access/dsl/core/types/standard.py +0 -92
- relationalai/early_access/dsl/core/types/unconstrained.py +0 -50
- relationalai/early_access/dsl/core/types/variables.py +0 -203
- relationalai/early_access/dsl/ir/compiler.py +0 -318
- relationalai/early_access/dsl/ir/executor.py +0 -260
- relationalai/early_access/dsl/ontologies/constraints.py +0 -88
- relationalai/early_access/dsl/ontologies/export.py +0 -30
- relationalai/early_access/dsl/ontologies/models.py +0 -453
- relationalai/early_access/dsl/ontologies/python_printer.py +0 -303
- relationalai/early_access/dsl/ontologies/readings.py +0 -60
- relationalai/early_access/dsl/ontologies/relationships.py +0 -322
- relationalai/early_access/dsl/ontologies/roles.py +0 -87
- relationalai/early_access/dsl/ontologies/subtyping.py +0 -55
- relationalai/early_access/dsl/orm/constraints.py +0 -438
- relationalai/early_access/dsl/orm/measures/dimensions.py +0 -200
- relationalai/early_access/dsl/orm/measures/initializer.py +0 -16
- relationalai/early_access/dsl/orm/measures/measure_rules.py +0 -275
- relationalai/early_access/dsl/orm/measures/measures.py +0 -299
- relationalai/early_access/dsl/orm/measures/role_exprs.py +0 -268
- relationalai/early_access/dsl/orm/models.py +0 -256
- relationalai/early_access/dsl/orm/object_oriented_printer.py +0 -344
- relationalai/early_access/dsl/orm/printer.py +0 -469
- relationalai/early_access/dsl/orm/reasoners.py +0 -480
- relationalai/early_access/dsl/orm/relations.py +0 -19
- relationalai/early_access/dsl/orm/relationships.py +0 -251
- relationalai/early_access/dsl/orm/types.py +0 -42
- relationalai/early_access/dsl/orm/utils.py +0 -79
- relationalai/early_access/dsl/orm/verb.py +0 -204
- relationalai/early_access/dsl/physical_metadata/tables.py +0 -133
- relationalai/early_access/dsl/relations.py +0 -170
- relationalai/early_access/dsl/rulesets.py +0 -69
- relationalai/early_access/dsl/schemas/__init__.py +0 -450
- relationalai/early_access/dsl/schemas/builder.py +0 -48
- relationalai/early_access/dsl/schemas/comp_names.py +0 -51
- relationalai/early_access/dsl/schemas/components.py +0 -203
- relationalai/early_access/dsl/schemas/contexts.py +0 -156
- relationalai/early_access/dsl/schemas/exprs.py +0 -89
- relationalai/early_access/dsl/schemas/fragments.py +0 -464
- relationalai/early_access/dsl/serialization.py +0 -79
- relationalai/early_access/dsl/serialize/exporter.py +0 -163
- relationalai/early_access/dsl/snow/api.py +0 -104
- relationalai/early_access/dsl/snow/common.py +0 -76
- relationalai/early_access/dsl/state_mgmt/__init__.py +0 -129
- relationalai/early_access/dsl/state_mgmt/state_charts.py +0 -125
- relationalai/early_access/dsl/state_mgmt/transitions.py +0 -130
- relationalai/early_access/dsl/types/__init__.py +0 -40
- relationalai/early_access/dsl/types/concepts.py +0 -12
- relationalai/early_access/dsl/types/entities.py +0 -135
- relationalai/early_access/dsl/types/values.py +0 -17
- relationalai/early_access/dsl/utils.py +0 -102
- relationalai/early_access/graphs/__init__.py +0 -13
- relationalai/early_access/lqp/__init__.py +0 -12
- relationalai/early_access/lqp/compiler/__init__.py +0 -12
- relationalai/early_access/lqp/constructors/__init__.py +0 -18
- relationalai/early_access/lqp/executor/__init__.py +0 -12
- relationalai/early_access/lqp/ir/__init__.py +0 -12
- relationalai/early_access/lqp/passes/__init__.py +0 -12
- relationalai/early_access/lqp/pragmas/__init__.py +0 -12
- relationalai/early_access/lqp/primitives/__init__.py +0 -12
- relationalai/early_access/lqp/types/__init__.py +0 -12
- relationalai/early_access/lqp/utils/__init__.py +0 -12
- relationalai/early_access/lqp/validators/__init__.py +0 -12
- relationalai/early_access/metamodel/__init__.py +0 -58
- relationalai/early_access/metamodel/builtins/__init__.py +0 -12
- relationalai/early_access/metamodel/compiler/__init__.py +0 -12
- relationalai/early_access/metamodel/dependency/__init__.py +0 -12
- relationalai/early_access/metamodel/factory/__init__.py +0 -17
- relationalai/early_access/metamodel/helpers/__init__.py +0 -12
- relationalai/early_access/metamodel/ir/__init__.py +0 -14
- relationalai/early_access/metamodel/rewrite/__init__.py +0 -7
- relationalai/early_access/metamodel/typer/__init__.py +0 -3
- relationalai/early_access/metamodel/typer/typer/__init__.py +0 -12
- relationalai/early_access/metamodel/types/__init__.py +0 -15
- relationalai/early_access/metamodel/util/__init__.py +0 -15
- relationalai/early_access/metamodel/visitor/__init__.py +0 -12
- relationalai/early_access/rel/__init__.py +0 -12
- relationalai/early_access/rel/executor/__init__.py +0 -12
- relationalai/early_access/rel/rel_utils/__init__.py +0 -12
- relationalai/early_access/rel/rewrite/__init__.py +0 -7
- relationalai/early_access/solvers/__init__.py +0 -19
- relationalai/early_access/sql/__init__.py +0 -11
- relationalai/early_access/sql/executor/__init__.py +0 -3
- relationalai/early_access/sql/rewrite/__init__.py +0 -3
- relationalai/early_access/tests/logging/__init__.py +0 -12
- relationalai/early_access/tests/test_snapshot_base/__init__.py +0 -12
- relationalai/early_access/tests/utils/__init__.py +0 -12
- relationalai/environments/__init__.py +0 -35
- relationalai/environments/base.py +0 -381
- relationalai/environments/colab.py +0 -14
- relationalai/environments/generic.py +0 -71
- relationalai/environments/ipython.py +0 -68
- relationalai/environments/jupyter.py +0 -9
- relationalai/environments/snowbook.py +0 -169
- relationalai/errors.py +0 -2455
- relationalai/experimental/SF.py +0 -38
- relationalai/experimental/inspect.py +0 -47
- relationalai/experimental/pathfinder/__init__.py +0 -158
- relationalai/experimental/pathfinder/api.py +0 -160
- relationalai/experimental/pathfinder/automaton.py +0 -584
- relationalai/experimental/pathfinder/bridge.py +0 -226
- relationalai/experimental/pathfinder/compiler.py +0 -416
- relationalai/experimental/pathfinder/datalog.py +0 -214
- relationalai/experimental/pathfinder/diagnostics.py +0 -56
- relationalai/experimental/pathfinder/filter.py +0 -236
- relationalai/experimental/pathfinder/glushkov.py +0 -439
- relationalai/experimental/pathfinder/options.py +0 -265
- relationalai/experimental/pathfinder/pathfinder-v0.7.0.rel +0 -1951
- relationalai/experimental/pathfinder/rpq.py +0 -344
- relationalai/experimental/pathfinder/transition.py +0 -200
- relationalai/experimental/pathfinder/utils.py +0 -26
- relationalai/experimental/paths/README.md +0 -107
- relationalai/experimental/paths/api.py +0 -143
- relationalai/experimental/paths/benchmarks/grid_graph.py +0 -37
- relationalai/experimental/paths/code_organization.md +0 -2
- relationalai/experimental/paths/examples/Movies.ipynb +0 -16328
- relationalai/experimental/paths/examples/basic_example.py +0 -40
- relationalai/experimental/paths/examples/minimal_engine_warmup.py +0 -3
- relationalai/experimental/paths/examples/movie_example.py +0 -77
- relationalai/experimental/paths/examples/movies_data/actedin.csv +0 -193
- relationalai/experimental/paths/examples/movies_data/directed.csv +0 -45
- relationalai/experimental/paths/examples/movies_data/follows.csv +0 -7
- relationalai/experimental/paths/examples/movies_data/movies.csv +0 -39
- relationalai/experimental/paths/examples/movies_data/person.csv +0 -134
- relationalai/experimental/paths/examples/movies_data/produced.csv +0 -16
- relationalai/experimental/paths/examples/movies_data/ratings.csv +0 -10
- relationalai/experimental/paths/examples/movies_data/wrote.csv +0 -11
- relationalai/experimental/paths/examples/paths_benchmark.py +0 -115
- relationalai/experimental/paths/examples/paths_example.py +0 -116
- relationalai/experimental/paths/examples/pattern_to_automaton.py +0 -28
- relationalai/experimental/paths/find_paths_via_automaton.py +0 -85
- relationalai/experimental/paths/graph.py +0 -185
- relationalai/experimental/paths/path_algorithms/find_paths.py +0 -280
- relationalai/experimental/paths/path_algorithms/one_sided_ball_repetition.py +0 -26
- relationalai/experimental/paths/path_algorithms/one_sided_ball_upto.py +0 -111
- relationalai/experimental/paths/path_algorithms/single.py +0 -59
- relationalai/experimental/paths/path_algorithms/two_sided_balls_repetition.py +0 -39
- relationalai/experimental/paths/path_algorithms/two_sided_balls_upto.py +0 -103
- relationalai/experimental/paths/path_algorithms/usp-old.py +0 -130
- relationalai/experimental/paths/path_algorithms/usp-tuple.py +0 -183
- relationalai/experimental/paths/path_algorithms/usp.py +0 -150
- relationalai/experimental/paths/product_graph.py +0 -93
- relationalai/experimental/paths/rpq/automaton.py +0 -584
- relationalai/experimental/paths/rpq/diagnostics.py +0 -56
- relationalai/experimental/paths/rpq/rpq.py +0 -378
- relationalai/experimental/paths/tests/tests_limit_sp_max_length.py +0 -90
- relationalai/experimental/paths/tests/tests_limit_sp_multiple.py +0 -119
- relationalai/experimental/paths/tests/tests_limit_sp_single.py +0 -104
- relationalai/experimental/paths/tests/tests_limit_walks_multiple.py +0 -113
- relationalai/experimental/paths/tests/tests_limit_walks_single.py +0 -149
- relationalai/experimental/paths/tests/tests_one_sided_ball_repetition_multiple.py +0 -70
- relationalai/experimental/paths/tests/tests_one_sided_ball_repetition_single.py +0 -64
- relationalai/experimental/paths/tests/tests_one_sided_ball_upto_multiple.py +0 -115
- relationalai/experimental/paths/tests/tests_one_sided_ball_upto_single.py +0 -75
- relationalai/experimental/paths/tests/tests_single_paths.py +0 -152
- relationalai/experimental/paths/tests/tests_single_walks.py +0 -208
- relationalai/experimental/paths/tests/tests_single_walks_undirected.py +0 -297
- relationalai/experimental/paths/tests/tests_two_sided_balls_repetition_multiple.py +0 -107
- relationalai/experimental/paths/tests/tests_two_sided_balls_repetition_single.py +0 -76
- relationalai/experimental/paths/tests/tests_two_sided_balls_upto_multiple.py +0 -76
- relationalai/experimental/paths/tests/tests_two_sided_balls_upto_single.py +0 -110
- relationalai/experimental/paths/tests/tests_usp_nsp_multiple.py +0 -229
- relationalai/experimental/paths/tests/tests_usp_nsp_single.py +0 -108
- relationalai/experimental/paths/tree_agg.py +0 -168
- relationalai/experimental/paths/utilities/iterators.py +0 -27
- relationalai/experimental/paths/utilities/prefix_sum.py +0 -91
- relationalai/experimental/solvers.py +0 -1087
- relationalai/loaders/csv.py +0 -195
- relationalai/loaders/loader.py +0 -177
- relationalai/loaders/types.py +0 -23
- relationalai/rel_emitter.py +0 -373
- relationalai/rel_utils.py +0 -185
- relationalai/semantics/designs/query_builder/identify_by.md +0 -106
- relationalai/semantics/devtools/benchmark_lqp.py +0 -536
- relationalai/semantics/devtools/compilation_manager.py +0 -294
- relationalai/semantics/devtools/extract_lqp.py +0 -110
- relationalai/semantics/internal/internal.py +0 -3785
- relationalai/semantics/internal/snowflake.py +0 -324
- relationalai/semantics/lqp/README.md +0 -34
- relationalai/semantics/lqp/builtins.py +0 -16
- relationalai/semantics/lqp/compiler.py +0 -22
- relationalai/semantics/lqp/constructors.py +0 -68
- relationalai/semantics/lqp/executor.py +0 -469
- relationalai/semantics/lqp/intrinsics.py +0 -24
- relationalai/semantics/lqp/model2lqp.py +0 -839
- relationalai/semantics/lqp/passes.py +0 -680
- relationalai/semantics/lqp/primitives.py +0 -252
- relationalai/semantics/lqp/result_helpers.py +0 -202
- relationalai/semantics/lqp/rewrite/annotate_constraints.py +0 -57
- relationalai/semantics/lqp/rewrite/cdc.py +0 -216
- relationalai/semantics/lqp/rewrite/extract_common.py +0 -338
- relationalai/semantics/lqp/rewrite/extract_keys.py +0 -449
- relationalai/semantics/lqp/rewrite/function_annotations.py +0 -114
- relationalai/semantics/lqp/rewrite/functional_dependencies.py +0 -314
- relationalai/semantics/lqp/rewrite/quantify_vars.py +0 -296
- relationalai/semantics/lqp/rewrite/splinter.py +0 -76
- relationalai/semantics/lqp/types.py +0 -101
- relationalai/semantics/lqp/utils.py +0 -160
- relationalai/semantics/lqp/validators.py +0 -57
- relationalai/semantics/metamodel/compiler.py +0 -133
- relationalai/semantics/metamodel/dependency.py +0 -862
- relationalai/semantics/metamodel/executor.py +0 -61
- relationalai/semantics/metamodel/factory.py +0 -287
- relationalai/semantics/metamodel/helpers.py +0 -361
- relationalai/semantics/metamodel/rewrite/discharge_constraints.py +0 -39
- relationalai/semantics/metamodel/rewrite/dnf_union_splitter.py +0 -210
- relationalai/semantics/metamodel/rewrite/extract_nested_logicals.py +0 -78
- relationalai/semantics/metamodel/rewrite/flatten.py +0 -549
- relationalai/semantics/metamodel/rewrite/format_outputs.py +0 -165
- relationalai/semantics/metamodel/typer/checker.py +0 -353
- relationalai/semantics/metamodel/typer/typer.py +0 -1395
- relationalai/semantics/reasoners/__init__.py +0 -10
- relationalai/semantics/reasoners/graph/README.md +0 -620
- relationalai/semantics/reasoners/graph/__init__.py +0 -37
- relationalai/semantics/reasoners/graph/core.py +0 -9020
- relationalai/semantics/reasoners/graph/design/beyond_demand_transform.md +0 -797
- relationalai/semantics/reasoners/graph/tests/README.md +0 -21
- relationalai/semantics/reasoners/optimization/__init__.py +0 -68
- relationalai/semantics/reasoners/optimization/common.py +0 -88
- relationalai/semantics/reasoners/optimization/solvers_dev.py +0 -568
- relationalai/semantics/reasoners/optimization/solvers_pb.py +0 -1163
- relationalai/semantics/rel/builtins.py +0 -40
- relationalai/semantics/rel/compiler.py +0 -989
- relationalai/semantics/rel/executor.py +0 -359
- relationalai/semantics/rel/rel.py +0 -482
- relationalai/semantics/rel/rel_utils.py +0 -276
- relationalai/semantics/snowflake/__init__.py +0 -3
- relationalai/semantics/sql/compiler.py +0 -2503
- relationalai/semantics/sql/executor/duck_db.py +0 -52
- relationalai/semantics/sql/executor/result_helpers.py +0 -64
- relationalai/semantics/sql/executor/snowflake.py +0 -145
- relationalai/semantics/sql/rewrite/denormalize.py +0 -222
- relationalai/semantics/sql/rewrite/double_negation.py +0 -49
- relationalai/semantics/sql/rewrite/recursive_union.py +0 -127
- relationalai/semantics/sql/rewrite/sort_output_query.py +0 -246
- relationalai/semantics/sql/sql.py +0 -504
- relationalai/semantics/std/pragmas.py +0 -11
- relationalai/semantics/std/std.py +0 -14
- relationalai/semantics/tests/test_snapshot_abstract.py +0 -143
- relationalai/semantics/tests/test_snapshot_base.py +0 -9
- relationalai/semantics/tests/utils.py +0 -46
- relationalai/std/__init__.py +0 -70
- relationalai/tools/cli.py +0 -1940
- relationalai/tools/cli_controls.py +0 -1826
- relationalai/tools/cli_helpers.py +0 -390
- relationalai/tools/debugger_client.py +0 -109
- relationalai/tools/debugger_server.py +0 -302
- relationalai/tools/dev.py +0 -685
- relationalai/tools/notes +0 -7
- relationalai/tools/qb_debugger.py +0 -425
- relationalai/util/clean_up_databases.py +0 -95
- relationalai/util/list_databases.py +0 -9
- relationalai/util/otel_configuration.py +0 -25
- relationalai/util/otel_handler.py +0 -484
- relationalai/util/snowflake_handler.py +0 -88
- relationalai/util/span_format_test.py +0 -43
- relationalai/util/span_tracker.py +0 -207
- relationalai/util/spans_file_handler.py +0 -72
- relationalai/util/tracing_handler.py +0 -34
- relationalai-0.12.13.dist-info/METADATA +0 -74
- relationalai-0.12.13.dist-info/RECORD +0 -449
- relationalai-0.12.13.dist-info/WHEEL +0 -4
- relationalai-0.12.13.dist-info/entry_points.txt +0 -3
- relationalai-0.12.13.dist-info/licenses/LICENSE +0 -202
- relationalai_test_util/__init__.py +0 -4
- relationalai_test_util/fixtures.py +0 -228
- relationalai_test_util/snapshot.py +0 -252
- relationalai_test_util/traceback.py +0 -118
- /relationalai/{analysis → semantics/frontend}/__init__.py +0 -0
- /relationalai/{auth/__init__.py → semantics/metamodel/metamodel_compiler.py} +0 -0
- /relationalai/{early_access → shims}/__init__.py +0 -0
- {relationalai/early_access/dsl/adapters → v0/relationalai/analysis}/__init__.py +0 -0
- {relationalai → v0/relationalai}/analysis/mechanistic.py +0 -0
- {relationalai → v0/relationalai}/analysis/whynot.py +0 -0
- {relationalai/early_access/dsl/adapters/orm → v0/relationalai/auth}/__init__.py +0 -0
- {relationalai → v0/relationalai}/auth/jwt_generator.py +0 -0
- {relationalai → v0/relationalai}/auth/oauth_callback_server.py +0 -0
- {relationalai → v0/relationalai}/auth/token_handler.py +0 -0
- {relationalai → v0/relationalai}/auth/util.py +0 -0
- {relationalai → v0/relationalai}/clients/__init__.py +0 -0
- {relationalai → v0/relationalai}/clients/cache_store.py +0 -0
- {relationalai → v0/relationalai}/compiler.py +0 -0
- {relationalai → v0/relationalai}/dependencies.py +0 -0
- {relationalai → v0/relationalai}/docutils.py +0 -0
- {relationalai/early_access/dsl/adapters/owl → v0/relationalai/early_access}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/__init__.py +0 -0
- {relationalai/early_access/dsl/bindings → v0/relationalai/early_access/dsl/adapters}/__init__.py +0 -0
- {relationalai/early_access/dsl/bindings/legacy → v0/relationalai/early_access/dsl/adapters/orm}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/adapters/orm/model.py +0 -0
- {relationalai/early_access/dsl/codegen → v0/relationalai/early_access/dsl/adapters/owl}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/adapters/owl/model.py +0 -0
- {relationalai/early_access/dsl/core/temporal → v0/relationalai/early_access/dsl/bindings}/__init__.py +0 -0
- {relationalai/early_access/dsl/ir → v0/relationalai/early_access/dsl/bindings/legacy}/__init__.py +0 -0
- {relationalai/early_access/dsl/ontologies → v0/relationalai/early_access/dsl/codegen}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/constants.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/core/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/core/constraints/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/core/constraints/predicate/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/core/stack.py +0 -0
- {relationalai/early_access/dsl/orm → v0/relationalai/early_access/dsl/core/temporal}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/core/utils.py +0 -0
- {relationalai/early_access/dsl/orm/measures → v0/relationalai/early_access/dsl/ir}/__init__.py +0 -0
- {relationalai/early_access/dsl/physical_metadata → v0/relationalai/early_access/dsl/ontologies}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/ontologies/raw_source.py +0 -0
- {relationalai/early_access/dsl/serialize → v0/relationalai/early_access/dsl/orm}/__init__.py +0 -0
- {relationalai/early_access/dsl/snow → v0/relationalai/early_access/dsl/orm/measures}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/orm/reasoner_errors.py +0 -0
- {relationalai/loaders → v0/relationalai/early_access/dsl/physical_metadata}/__init__.py +0 -0
- {relationalai/semantics/tests → v0/relationalai/early_access/dsl/serialize}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/serialize/binding_model.py +0 -0
- {relationalai → v0/relationalai}/early_access/dsl/serialize/model.py +0 -0
- {relationalai/tools → v0/relationalai/early_access/dsl/snow}/__init__.py +0 -0
- {relationalai → v0/relationalai}/early_access/tests/__init__.py +0 -0
- {relationalai → v0/relationalai}/environments/ci.py +0 -0
- {relationalai → v0/relationalai}/environments/hex.py +0 -0
- {relationalai → v0/relationalai}/environments/terminal.py +0 -0
- {relationalai → v0/relationalai}/experimental/__init__.py +0 -0
- {relationalai → v0/relationalai}/experimental/graphs.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/__init__.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/benchmarks/__init__.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/path_algorithms/__init__.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/rpq/__init__.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/rpq/filter.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/rpq/glushkov.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/rpq/transition.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/utilities/__init__.py +0 -0
- {relationalai → v0/relationalai}/experimental/paths/utilities/utilities.py +0 -0
- {relationalai → v0/relationalai}/metagen.py +0 -0
- {relationalai → v0/relationalai}/metamodel.py +0 -0
- {relationalai → v0/relationalai}/rel.py +0 -0
- {relationalai → v0/relationalai}/semantics/devtools/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/internal/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/internal/annotations.py +0 -0
- {relationalai → v0/relationalai}/semantics/lqp/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/lqp/ir.py +0 -0
- {relationalai → v0/relationalai}/semantics/lqp/pragmas.py +0 -0
- {relationalai → v0/relationalai}/semantics/lqp/rewrite/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/metamodel/dataflow.py +0 -0
- {relationalai → v0/relationalai}/semantics/metamodel/ir.py +0 -0
- {relationalai → v0/relationalai}/semantics/metamodel/rewrite/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/metamodel/typer/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/metamodel/types.py +0 -0
- {relationalai → v0/relationalai}/semantics/metamodel/util.py +0 -0
- {relationalai → v0/relationalai}/semantics/metamodel/visitor.py +0 -0
- {relationalai → v0/relationalai}/semantics/reasoners/experimental/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/rel/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/sql/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/sql/executor/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/sql/rewrite/__init__.py +0 -0
- {relationalai → v0/relationalai}/semantics/tests/logging.py +0 -0
- {relationalai → v0/relationalai}/std/aggregates.py +0 -0
- {relationalai → v0/relationalai}/std/dates.py +0 -0
- {relationalai → v0/relationalai}/std/graphs.py +0 -0
- {relationalai → v0/relationalai}/std/inspect.py +0 -0
- {relationalai → v0/relationalai}/std/math.py +0 -0
- {relationalai → v0/relationalai}/std/re.py +0 -0
- {relationalai → v0/relationalai}/std/strings.py +0 -0
- {relationalai → v0/relationalai}/tools/cleanup_snapshots.py +0 -0
- {relationalai → v0/relationalai}/tools/constants.py +0 -0
- {relationalai → v0/relationalai}/tools/query_utils.py +0 -0
- {relationalai → v0/relationalai}/tools/snapshot_viewer.py +0 -0
- {relationalai → v0/relationalai}/util/__init__.py +0 -0
- {relationalai → v0/relationalai}/util/constants.py +0 -0
- {relationalai → v0/relationalai}/util/graph.py +0 -0
- {relationalai → v0/relationalai}/util/timeout.py +0 -0
|
@@ -1,130 +0,0 @@
|
|
|
1
|
-
# For builder components.
|
|
2
|
-
from relationalai.semantics import Integer, define, max, sum, not_
|
|
3
|
-
from relationalai.experimental.paths.graph import Graph
|
|
4
|
-
from relationalai.experimental.paths.path_algorithms.one_sided_ball_upto import ball_upto
|
|
5
|
-
from relationalai.experimental.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
def compute_usp(g: Graph, Source, Target, max_length = None):
|
|
9
|
-
# Computes the Union of Shortest Paths (USP) from a source set to a destination set
|
|
10
|
-
# USP is a subgraph of the original graph
|
|
11
|
-
|
|
12
|
-
edge = g.Edge
|
|
13
|
-
Node = g.Node
|
|
14
|
-
|
|
15
|
-
n, d = Integer.ref(), Integer.ref()
|
|
16
|
-
tgt, u, v, w = Node.ref(), g.Node.ref(), g.Node.ref(), g.Node.ref()
|
|
17
|
-
|
|
18
|
-
if max_length is None:
|
|
19
|
-
ball = ball_upto(g, Source, Target)
|
|
20
|
-
usp_nodes = g.model.Relationship(f"Node {{{Node}}} is in the USP")
|
|
21
|
-
usp = g.model.Relationship(f"Edge {{{Node}}} to {{{Node}}} is in the USP")
|
|
22
|
-
boundary = g.model.Relationship(f"Target node {{{Node}}} is in the boundary of the ball")
|
|
23
|
-
|
|
24
|
-
# propagate backwards to find the nodes that are in the USP:
|
|
25
|
-
define(boundary(tgt)).where(
|
|
26
|
-
Target(tgt),
|
|
27
|
-
ball(n, tgt)
|
|
28
|
-
)
|
|
29
|
-
|
|
30
|
-
define(usp_nodes(tgt)).where(
|
|
31
|
-
boundary(tgt)
|
|
32
|
-
)
|
|
33
|
-
|
|
34
|
-
define(usp_nodes(u)).where(
|
|
35
|
-
usp(u, v)
|
|
36
|
-
)
|
|
37
|
-
|
|
38
|
-
define(usp(u, v)).where(
|
|
39
|
-
usp_nodes(v),
|
|
40
|
-
ball(d, v),
|
|
41
|
-
ball(d - 1, u),
|
|
42
|
-
edge(u, v)
|
|
43
|
-
)
|
|
44
|
-
else:
|
|
45
|
-
ball = ball_with_repetition(g, Source, max_length)
|
|
46
|
-
max_dist = g.model.Relationship("The maximum distance of a target node form the source nodes is {Integer}")
|
|
47
|
-
usp = g.model.Relationship(f"{{{Node}}} at distance {{Integer}} from the source nodes is connected with {{{Node}}}")
|
|
48
|
-
boundary = g.model.Relationship(f"Target pair {{{Node}}} {{Integer}} is in the boundary of the ball")
|
|
49
|
-
|
|
50
|
-
define(max_dist(max(n))).where(
|
|
51
|
-
ball(n, tgt),
|
|
52
|
-
Target(tgt)
|
|
53
|
-
)
|
|
54
|
-
|
|
55
|
-
define(boundary(tgt, n)).where(
|
|
56
|
-
max_dist(n),
|
|
57
|
-
Target(tgt),
|
|
58
|
-
ball(n, tgt)
|
|
59
|
-
)
|
|
60
|
-
|
|
61
|
-
define(usp(u, n, tgt)).where(
|
|
62
|
-
boundary(tgt, n + 1),
|
|
63
|
-
ball(n, u),
|
|
64
|
-
edge(u, v)
|
|
65
|
-
)
|
|
66
|
-
|
|
67
|
-
define(usp(u, n, v)).where(
|
|
68
|
-
usp(v, n + 1, w),
|
|
69
|
-
ball(n, u),
|
|
70
|
-
edge(u, v)
|
|
71
|
-
)
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
return usp, boundary
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
def compute_nsp_from_usp(g: Graph, usp, Source, Target, Boundary):
|
|
80
|
-
Node = g.Node
|
|
81
|
-
|
|
82
|
-
n, m = Integer.ref(), Integer.ref()
|
|
83
|
-
tgt, u, v = Node.ref(), Node.ref(), Node.ref()
|
|
84
|
-
|
|
85
|
-
if usp._arity() == 2:
|
|
86
|
-
nsp = g.model.Relationship(f"Number of shortest paths from {{{Node}}} to destination is {{Integer}}")
|
|
87
|
-
|
|
88
|
-
define(nsp(tgt, 1)).where(
|
|
89
|
-
Boundary(tgt)
|
|
90
|
-
)
|
|
91
|
-
|
|
92
|
-
define(nsp(tgt, 1)).where(
|
|
93
|
-
Target(tgt),
|
|
94
|
-
Source(tgt)
|
|
95
|
-
)
|
|
96
|
-
|
|
97
|
-
define(nsp(u, sum(v, n).per(u))).where(
|
|
98
|
-
nsp(v, n),
|
|
99
|
-
usp(u, v)
|
|
100
|
-
)
|
|
101
|
-
else:
|
|
102
|
-
nsp = g.model.Relationship(f"Number of shortest paths from pair {{{Node}}} {{Integer}} to destination is {{Integer}}")
|
|
103
|
-
|
|
104
|
-
define(nsp(tgt, n, 1)).where(
|
|
105
|
-
Boundary(tgt, n)
|
|
106
|
-
)
|
|
107
|
-
|
|
108
|
-
define(nsp(u, n, sum(v, n + 1, m).per(u, n))).where(
|
|
109
|
-
nsp(v, n + 1, m),
|
|
110
|
-
usp(u, n, v),
|
|
111
|
-
not_(Target(u))
|
|
112
|
-
)
|
|
113
|
-
|
|
114
|
-
define(nsp(u, n, 1 + sum(v, n + 1, m).per(u, n))).where(
|
|
115
|
-
nsp(v, n + 1, m),
|
|
116
|
-
usp(u, n, v),
|
|
117
|
-
Target(u),
|
|
118
|
-
not_(Boundary(u, n))
|
|
119
|
-
)
|
|
120
|
-
|
|
121
|
-
return nsp
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
def compute_nsp(g: Graph, Source, Target, max_length = None):
|
|
125
|
-
# Computes the number of shortest paths (NSP) from a source set to a destination set
|
|
126
|
-
|
|
127
|
-
usp, Boundary = compute_usp(g, Source, Target, max_length)
|
|
128
|
-
nsp = compute_nsp_from_usp(g, usp, Source, Target, Boundary)
|
|
129
|
-
|
|
130
|
-
return nsp
|
|
@@ -1,183 +0,0 @@
|
|
|
1
|
-
# For builder components.
|
|
2
|
-
from relationalai.semantics import Model, Integer, define, max, sum, select, not_
|
|
3
|
-
from relationalai.experimental.paths.graph import Graph
|
|
4
|
-
from relationalai.experimental.paths.path_algorithms.one_sided_ball_upto import ball_upto
|
|
5
|
-
from relationalai.experimental.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
def compute_usp(g: Graph, Source, Target, max_length = None):
|
|
9
|
-
# Computes the Union of Shortest Paths (USP) from a source set to a destination set
|
|
10
|
-
# USP is a subgraph of the original graph
|
|
11
|
-
|
|
12
|
-
edge = g.Edge
|
|
13
|
-
Node = g.Node
|
|
14
|
-
|
|
15
|
-
n, d = Integer.ref(), Integer.ref()
|
|
16
|
-
tgt, u, v = Node.ref(), g.Node.ref(), g.Node.ref()
|
|
17
|
-
|
|
18
|
-
if max_length is None:
|
|
19
|
-
ball = ball_upto(g, Source, Target)
|
|
20
|
-
usp_nodes = g.model.Relationship(f"usp_nodes1 {{{Node}}}")
|
|
21
|
-
usp = g.model.Relationship(f"usp1 {{{Node}}} {{{Node}}}")
|
|
22
|
-
boundary = g.model.Relationship(f"boundary1 {{{Node}}}")
|
|
23
|
-
|
|
24
|
-
# propagate backwards to find the nodes that are in the USP:
|
|
25
|
-
define(boundary(tgt)).where(
|
|
26
|
-
Target(tgt),
|
|
27
|
-
ball(n, tgt)
|
|
28
|
-
)
|
|
29
|
-
|
|
30
|
-
define(usp_nodes(tgt)).where(
|
|
31
|
-
boundary(tgt)
|
|
32
|
-
)
|
|
33
|
-
|
|
34
|
-
define(usp_nodes(u)).where(
|
|
35
|
-
usp(u, v)
|
|
36
|
-
)
|
|
37
|
-
|
|
38
|
-
define(usp(u, v)).where(
|
|
39
|
-
usp_nodes(v),
|
|
40
|
-
ball(d, v),
|
|
41
|
-
ball(d - 1, u),
|
|
42
|
-
edge(u, v)
|
|
43
|
-
)
|
|
44
|
-
else:
|
|
45
|
-
ball = ball_with_repetition(g, Source, max_length)
|
|
46
|
-
max_dist = g.model.Relationship("max_dist1 {Integer}")
|
|
47
|
-
NodeDist = g.model.Concept("NodeDist")
|
|
48
|
-
usp = g.model.Relationship(f"usp2 {{{NodeDist}}} {{{NodeDist}}}")
|
|
49
|
-
boundary = g.model.Relationship(f"boundary2 {{{NodeDist}}}")
|
|
50
|
-
|
|
51
|
-
define(NodeDist.new(node = u, layer = n)).where(
|
|
52
|
-
ball(n, u)
|
|
53
|
-
)
|
|
54
|
-
|
|
55
|
-
define(max_dist(max(n))).where(
|
|
56
|
-
ball(n, tgt),
|
|
57
|
-
Target(tgt)
|
|
58
|
-
)
|
|
59
|
-
|
|
60
|
-
uu, vv, ww = NodeDist.ref(), NodeDist.ref(), NodeDist.ref()
|
|
61
|
-
|
|
62
|
-
define(boundary(uu)).where(
|
|
63
|
-
NodeDist(uu),
|
|
64
|
-
max_dist(n),
|
|
65
|
-
Target(tgt),
|
|
66
|
-
ball(n, tgt),
|
|
67
|
-
uu.node == tgt,
|
|
68
|
-
uu.layer == n
|
|
69
|
-
)
|
|
70
|
-
|
|
71
|
-
define(usp(uu, vv)).where(
|
|
72
|
-
boundary(vv),
|
|
73
|
-
NodeDist(uu),
|
|
74
|
-
vv.node == v,
|
|
75
|
-
vv.layer == n + 1,
|
|
76
|
-
uu.node == u,
|
|
77
|
-
uu.layer == n,
|
|
78
|
-
ball(n, u),
|
|
79
|
-
edge(u, v)
|
|
80
|
-
)
|
|
81
|
-
|
|
82
|
-
define(usp(uu, vv)).where(
|
|
83
|
-
usp(vv, ww),
|
|
84
|
-
vv.node == v,
|
|
85
|
-
vv.layer == n + 1,
|
|
86
|
-
uu.node == u,
|
|
87
|
-
uu.layer == n,
|
|
88
|
-
ball(n, u),
|
|
89
|
-
edge(u, v)
|
|
90
|
-
)
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
return usp, boundary
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
def compute_nsp_from_usp(g: Graph, usp, Source, Target, Boundary):
|
|
99
|
-
Node = g.Node
|
|
100
|
-
|
|
101
|
-
n, m = Integer.ref(), Integer.ref()
|
|
102
|
-
tgt, u, v = Node.ref(), Node.ref(), Node.ref()
|
|
103
|
-
|
|
104
|
-
"""if max_dist is None:
|
|
105
|
-
nsp = g.model.Relationship(f"nsp2 {{{Node}}} {{Integer}}")
|
|
106
|
-
define(nsp(tgt, 1)).where(
|
|
107
|
-
Target(tgt),
|
|
108
|
-
usp(u, tgt)
|
|
109
|
-
)
|
|
110
|
-
|
|
111
|
-
define(nsp(tgt, 1)).where(
|
|
112
|
-
Target(tgt),
|
|
113
|
-
Source(tgt)
|
|
114
|
-
)
|
|
115
|
-
|
|
116
|
-
define(nsp(u, sum(v, n).per(u))).where(
|
|
117
|
-
nsp(v, n),
|
|
118
|
-
usp(u, v)
|
|
119
|
-
)
|
|
120
|
-
else:
|
|
121
|
-
"""
|
|
122
|
-
nsp = g.model.Relationship(f"nsp1 {{{Node}}} {{Integer}} {{Integer}}")
|
|
123
|
-
|
|
124
|
-
define(nsp(tgt, n, 1)).where(
|
|
125
|
-
Boundary(tgt, n)
|
|
126
|
-
)
|
|
127
|
-
|
|
128
|
-
define(nsp(u, n, sum(v, n + 1, m).per(u, n))).where(
|
|
129
|
-
nsp(v, n + 1, m),
|
|
130
|
-
usp(u, n, v),
|
|
131
|
-
not_(Target(u))
|
|
132
|
-
)
|
|
133
|
-
|
|
134
|
-
define(nsp(u, n, 1 + sum(v, n + 1, m).per(u, n))).where(
|
|
135
|
-
nsp(v, n + 1, m),
|
|
136
|
-
usp(u, n, v),
|
|
137
|
-
Target(u),
|
|
138
|
-
not_(Boundary(u, n))
|
|
139
|
-
)
|
|
140
|
-
|
|
141
|
-
return nsp
|
|
142
|
-
|
|
143
|
-
|
|
144
|
-
def compute_nsp(g: Graph, Source, Target, max_length = None):
|
|
145
|
-
# Computes the number of shortest paths (NSP) from a source set to a destination set
|
|
146
|
-
|
|
147
|
-
if max_length is None:
|
|
148
|
-
usp, Boundary = compute_usp(g, Source, Target)
|
|
149
|
-
nsp = compute_nsp_from_usp(g, usp, Source, Target, Boundary)
|
|
150
|
-
else:
|
|
151
|
-
usp, Boundary = compute_usp(g, Source, Target, max_length)
|
|
152
|
-
nsp = compute_nsp_from_usp(g, usp, Source, Target, Boundary)
|
|
153
|
-
|
|
154
|
-
return nsp
|
|
155
|
-
|
|
156
|
-
|
|
157
|
-
model = Model("test_usp_nsp_single", dry_run=False)
|
|
158
|
-
|
|
159
|
-
grid1 = Graph.construct_grid(model, 2, "undirected")
|
|
160
|
-
|
|
161
|
-
source1 = grid1.Node.new(row = 1, col = 1)
|
|
162
|
-
target1 = grid1.Node.new(row = 2, col = 2)
|
|
163
|
-
|
|
164
|
-
Source1 = model.Concept("Source1", extends=[grid1.Node])
|
|
165
|
-
Target1 = model.Concept("Target1", extends=[grid1.Node])
|
|
166
|
-
|
|
167
|
-
define(Source1(source1))
|
|
168
|
-
define(Target1(target1))
|
|
169
|
-
define(Target1(source1))
|
|
170
|
-
|
|
171
|
-
usp, boundary = compute_usp(grid1, Source1, Target1, 4)
|
|
172
|
-
|
|
173
|
-
usp.inspect()
|
|
174
|
-
#nsp = compute_nsp(grid1, Source1, Target1, 4)
|
|
175
|
-
|
|
176
|
-
|
|
177
|
-
u, v = grid1.Node.ref(), grid1.Node.ref(),
|
|
178
|
-
n, m = Integer.ref(), Integer.ref()
|
|
179
|
-
|
|
180
|
-
select(u, v).where(usp(u, v)).inspect()
|
|
181
|
-
|
|
182
|
-
|
|
183
|
-
# select(u.row, u.col, n, m).where(nsp(u, n, m)).inspect()
|
|
@@ -1,150 +0,0 @@
|
|
|
1
|
-
# For builder components.
|
|
2
|
-
from relationalai.semantics import Integer, define, sum, not_
|
|
3
|
-
from relationalai.experimental.paths.graph import Graph
|
|
4
|
-
from relationalai.experimental.paths.path_algorithms.one_sided_ball_upto import ball_upto
|
|
5
|
-
from relationalai.experimental.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
def compute_usp(g: Graph, Source, Target, max_length=None):
|
|
9
|
-
# Computes the Union of Shortest Paths (USP) from a source set to a destination set
|
|
10
|
-
# USP is a subgraph of the original graph
|
|
11
|
-
|
|
12
|
-
edge = g.Edge
|
|
13
|
-
Node = g.Node
|
|
14
|
-
|
|
15
|
-
n, d = Integer.ref(), Integer.ref()
|
|
16
|
-
tgt, u, v = Node.ref(), g.Node.ref(), g.Node.ref()
|
|
17
|
-
|
|
18
|
-
ball = ball_upto(g, Source, Target, max_length)
|
|
19
|
-
usp_nodes = g.model.Relationship(f"usp_nodes {{{Node}}}")
|
|
20
|
-
usp = g.model.Relationship(f"usp {{{Node}}} {{{Node}}}")
|
|
21
|
-
boundary = g.model.Relationship(f"usp_boundary {{{Node}}}")
|
|
22
|
-
|
|
23
|
-
# propagate backwards to find the nodes that are in the USP:
|
|
24
|
-
define(boundary(tgt)).where(
|
|
25
|
-
Target(tgt),
|
|
26
|
-
ball(n, tgt)
|
|
27
|
-
)
|
|
28
|
-
|
|
29
|
-
define(usp_nodes(tgt)).where(
|
|
30
|
-
boundary(tgt)
|
|
31
|
-
)
|
|
32
|
-
|
|
33
|
-
define(usp_nodes(u)).where(
|
|
34
|
-
usp(u, v)
|
|
35
|
-
)
|
|
36
|
-
|
|
37
|
-
define(usp(u, v)).where(
|
|
38
|
-
usp_nodes(v),
|
|
39
|
-
ball(d, v),
|
|
40
|
-
ball(d - 1, u),
|
|
41
|
-
edge(u, v)
|
|
42
|
-
)
|
|
43
|
-
|
|
44
|
-
return usp, boundary
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
def compute_uw(g: Graph, Source, Target, max_length):
|
|
48
|
-
# Computes the Union of Walks (UW) up to a given length from a source set to a destination set
|
|
49
|
-
|
|
50
|
-
edge = g.Edge
|
|
51
|
-
Node = g.Node
|
|
52
|
-
|
|
53
|
-
n = Integer.ref()
|
|
54
|
-
tgt, u, v, w = Node.ref(), g.Node.ref(), g.Node.ref(), g.Node.ref()
|
|
55
|
-
|
|
56
|
-
ball = ball_with_repetition(g, Source, max_length)
|
|
57
|
-
uw = g.model.Relationship(f"uw {{{Node}}} {{Integer}} {{{Node}}}")
|
|
58
|
-
boundary = g.model.Relationship(f"boundary {{{Node}}} {{Integer}}")
|
|
59
|
-
|
|
60
|
-
define(uw(u, n, tgt)).where(
|
|
61
|
-
Target(tgt),
|
|
62
|
-
ball(n + 1, tgt),
|
|
63
|
-
ball(n, u),
|
|
64
|
-
edge(u, tgt)
|
|
65
|
-
)
|
|
66
|
-
|
|
67
|
-
define(uw(u, n, v)).where(
|
|
68
|
-
uw(v, n + 1, w),
|
|
69
|
-
ball(n, u),
|
|
70
|
-
edge(u, v)
|
|
71
|
-
)
|
|
72
|
-
|
|
73
|
-
define(boundary(tgt, n)).where(
|
|
74
|
-
Target(tgt),
|
|
75
|
-
ball(n, tgt),
|
|
76
|
-
not_(uw(tgt, n, u))
|
|
77
|
-
)
|
|
78
|
-
|
|
79
|
-
return uw, boundary
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
def compute_nsp_from_usp(g: Graph, usp, Source, Target, Boundary):
|
|
83
|
-
Node = g.Node
|
|
84
|
-
|
|
85
|
-
n = Integer.ref()
|
|
86
|
-
tgt, u, v = Node.ref(), Node.ref(), Node.ref()
|
|
87
|
-
|
|
88
|
-
nsp = g.model.Relationship(f"num_shortest {{{Node}}} {{Integer}}")
|
|
89
|
-
|
|
90
|
-
define(nsp(tgt, 1)).where(
|
|
91
|
-
Boundary(tgt)
|
|
92
|
-
)
|
|
93
|
-
|
|
94
|
-
define(nsp(tgt, 1)).where(
|
|
95
|
-
Target(tgt),
|
|
96
|
-
Source(tgt)
|
|
97
|
-
)
|
|
98
|
-
|
|
99
|
-
define(nsp(u, sum(v, n).per(u))).where(
|
|
100
|
-
nsp(v, n),
|
|
101
|
-
usp(u, v)
|
|
102
|
-
)
|
|
103
|
-
|
|
104
|
-
return nsp
|
|
105
|
-
|
|
106
|
-
|
|
107
|
-
def compute_nw_from_uw(g: Graph, usp, Target, Boundary):
|
|
108
|
-
Node = g.Node
|
|
109
|
-
|
|
110
|
-
n, m = Integer.ref(), Integer.ref()
|
|
111
|
-
tgt, u, v = Node.ref(), Node.ref(), Node.ref()
|
|
112
|
-
|
|
113
|
-
nw = g.model.Relationship(f"num_walks {{{Node}}} {{Integer}} {{Integer}}")
|
|
114
|
-
|
|
115
|
-
define(nw(tgt, n, 1)).where(
|
|
116
|
-
Boundary(tgt, n)
|
|
117
|
-
)
|
|
118
|
-
|
|
119
|
-
define(nw(u, n, sum(v, n + 1, m).per(u, n))).where(
|
|
120
|
-
nw(v, n + 1, m),
|
|
121
|
-
usp(u, n, v),
|
|
122
|
-
not_(Target(u))
|
|
123
|
-
)
|
|
124
|
-
|
|
125
|
-
define(nw(u, n, 1 + sum(v, n + 1, m).per(u, n))).where(
|
|
126
|
-
nw(v, n + 1, m),
|
|
127
|
-
usp(u, n, v),
|
|
128
|
-
Target(u),
|
|
129
|
-
not_(Boundary(u, n))
|
|
130
|
-
)
|
|
131
|
-
|
|
132
|
-
return nw
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
def compute_nsp(g: Graph, Source, Target):
|
|
136
|
-
# Computes the number of shortest paths (NSP) from a source set to a destination set
|
|
137
|
-
|
|
138
|
-
usp, Boundary = compute_usp(g, Source, Target)
|
|
139
|
-
nsp = compute_nsp_from_usp(g, usp, Source, Target, Boundary)
|
|
140
|
-
|
|
141
|
-
return nsp
|
|
142
|
-
|
|
143
|
-
|
|
144
|
-
def compute_nw(g: Graph, Source, Target, max_length):
|
|
145
|
-
# Computes the number of shortest paths (NSP) from a source set to a destination set
|
|
146
|
-
|
|
147
|
-
uw, Boundary = compute_uw(g, Source, Target, max_length)
|
|
148
|
-
nw = compute_nw_from_uw(g, uw, Target, Boundary)
|
|
149
|
-
|
|
150
|
-
return nw
|
|
@@ -1,93 +0,0 @@
|
|
|
1
|
-
"""
|
|
2
|
-
Product graph implementation for automaton-based path finding.
|
|
3
|
-
"""
|
|
4
|
-
|
|
5
|
-
from relationalai.semantics import Integer, where, define, Relationship, String
|
|
6
|
-
|
|
7
|
-
from .graph import Graph
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
def create_product_graph(graph, automaton):
|
|
11
|
-
"""
|
|
12
|
-
Create a product graph based on an automaton specification.
|
|
13
|
-
|
|
14
|
-
Args:
|
|
15
|
-
model: The RelationalAI model
|
|
16
|
-
automaton_spec: List of automaton transitions (src_state, dst_state, label/filter)
|
|
17
|
-
node: Node concept from the original graph
|
|
18
|
-
edge: Edge relationship from the original graph
|
|
19
|
-
|
|
20
|
-
Returns:
|
|
21
|
-
Tuple of (ProductGraphNode, product_graph) representing the product graph concepts
|
|
22
|
-
"""
|
|
23
|
-
model = graph.model
|
|
24
|
-
Node = graph.Node
|
|
25
|
-
Edge = graph.Edge
|
|
26
|
-
|
|
27
|
-
# Define product graph for path finding
|
|
28
|
-
ProductGraphNode = model.Concept("ProductNode")
|
|
29
|
-
ProductGraphEdge = model.Relationship(f"pg_edge {{{ProductGraphNode}}} {{{ProductGraphNode}}}")
|
|
30
|
-
|
|
31
|
-
ProductGraphNode.orig_node = Relationship(f"orig_node {{{ProductGraphNode}}} {{{Node}}}")
|
|
32
|
-
ProductGraphNode.state = Relationship(f"state {{{ProductGraphNode}}} {{Integer}}")
|
|
33
|
-
ProductGraphNode.identify_by(ProductGraphNode.state, ProductGraphNode.orig_node)
|
|
34
|
-
|
|
35
|
-
# NB: Modifying `automaton` object
|
|
36
|
-
# Add automaton transitions to the model
|
|
37
|
-
automaton.transition = model.Relationship("automaton {Integer} {Integer} {String}")
|
|
38
|
-
|
|
39
|
-
# Create references for use in rules
|
|
40
|
-
state1, state2 = Integer.ref(), Integer.ref()
|
|
41
|
-
node1, node2 = Node.ref(), Node.ref()
|
|
42
|
-
|
|
43
|
-
# Create a rule for each automaton transition:
|
|
44
|
-
for state1 in automaton.delta: # delta is the dict containing automaton transitions
|
|
45
|
-
for state2 in automaton.delta[state1]:
|
|
46
|
-
for transition in automaton.delta[state1][state2]:
|
|
47
|
-
label = transition.get_label().label
|
|
48
|
-
|
|
49
|
-
define(automaton.transition(state1, state2, label))
|
|
50
|
-
|
|
51
|
-
source_filters = transition.src_filters
|
|
52
|
-
|
|
53
|
-
where(
|
|
54
|
-
ProductGraphNode.state == state1,
|
|
55
|
-
ProductGraphNode.orig_node == node1,
|
|
56
|
-
Edge(node1, node2, label),
|
|
57
|
-
*([f(node1) for f in source_filters])
|
|
58
|
-
).define(
|
|
59
|
-
ProductGraphNode.new(state=state2, orig_node=node2),
|
|
60
|
-
ProductGraphEdge(ProductGraphNode, ProductGraphNode.new(state=state2, orig_node=node2))
|
|
61
|
-
)
|
|
62
|
-
|
|
63
|
-
return Graph(model, ProductGraphNode, ProductGraphEdge)
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
def project_product_graph_paths_to_original_graph(product_graph, product_graph_paths, automaton):
|
|
67
|
-
|
|
68
|
-
model = product_graph.model
|
|
69
|
-
ProductGraphNode = product_graph.Node
|
|
70
|
-
|
|
71
|
-
# extract Node type of original graph:
|
|
72
|
-
Node = ProductGraphNode.orig_node._fields[1].type_str
|
|
73
|
-
|
|
74
|
-
# print("Node type in original graph:", Node)
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
# Project path down from product graph to original graph
|
|
78
|
-
projected_paths = model.Relationship(f"projected {{Integer}} {{Integer}} {{String}} {{{Node}}}")
|
|
79
|
-
|
|
80
|
-
k = Integer.ref()
|
|
81
|
-
product_graph_node2 = ProductGraphNode.ref()
|
|
82
|
-
label = String.ref()
|
|
83
|
-
|
|
84
|
-
i = Integer.ref()
|
|
85
|
-
|
|
86
|
-
define(projected_paths(i, 0, "", ProductGraphNode.orig_node)).where(product_graph_paths(i, 0, ProductGraphNode))
|
|
87
|
-
define(projected_paths(i, k, label, product_graph_node2.orig_node)).where(
|
|
88
|
-
product_graph_paths(i, k - 1, ProductGraphNode),
|
|
89
|
-
product_graph_paths(i, k, product_graph_node2),
|
|
90
|
-
automaton.transition(ProductGraphNode.state, product_graph_node2.state, label)
|
|
91
|
-
)
|
|
92
|
-
|
|
93
|
-
return projected_paths
|